SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 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
1 changed files with 3 additions and 0 deletions
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 torch_device
) )
if torch_device == "cuda":
rag_token.half()
input_dict = tokenizer( input_dict = tokenizer(
self.test_data_questions, self.test_data_questions,
return_tensors="pt", return_tensors="pt",