riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

[tests] make cuda-only tests device-agnostic (#35222)

Browse Source 
fix cuda-only tests
This commit is contained in:
Fanli Lin 2024-12-18 17:14:22 +08:00 committed by GitHub
parent 1eee1cedfd
commit c7e48053aa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/models/rag/test_modeling_rag.py
View File

@ -33,7 +33,7 @@ from transformers.testing_utils import (
require_sentencepiece,
require_tokenizers,
require_torch,
require_torch_non_multi_gpu,
require_torch_non_multi_accelerator,
slow,
torch_device,
)
@ -678,7 +678,7 @@ class RagDPRT5Test(RagTestMixin, unittest.TestCase):
@require_retrieval
@require_sentencepiece
@require_tokenizers
@require_torch_non_multi_gpu
@require_torch_non_multi_accelerator
class RagModelIntegrationTests(unittest.TestCase):
def tearDown(self):
super().tearDown()
@ -1002,7 +1002,7 @@ class RagModelIntegrationTests(unittest.TestCase):
torch_device
)
if torch_device == "cuda":
if torch_device != "cpu":
rag_token.half()
input_dict = tokenizer(