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

correct (#13304)

Browse Source
This commit is contained in:
Patrick von Platen 2021-08-30 12:02:08 +02:00 committed by GitHub
parent 4046e66e40
commit 4362ee298a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_modeling_rag.py
View File

@ -988,6 +988,9 @@ class RagModelIntegrationTests(unittest.TestCase):
torch_device
)
if torch_device == "cuda":
rag_token.half()
input_dict = tokenizer(
self.test_data_questions,
return_tensors="pt",