correct (huggingface#13304)

patrickvonplaten · web-flow · commit 4362ee298a92 · 2021-08-30T06:02:08.000-04:00
diff --git a/tests/test_modeling_rag.py b/tests/test_modeling_rag.py
@@ -988,6 +988,9 @@ def test_rag_token_generate_batch(self):
             torch_device
         )
 
+        if torch_device == "cuda":
+            rag_token.half()
+
         input_dict = tokenizer(
             self.test_data_questions,
             return_tensors="pt",

Original file line number	Diff line number	Diff line change
`@@ -988,6 +988,9 @@ def test_rag_token_generate_batch(self):`
`988`	`988`	`torch_device`
`989`	`989`	`)`
`990`	`990`
	`991`	`+ if torch_device == "cuda":`
	`992`	`+ rag_token.half()`
	`993`	`+`
`991`	`994`	`input_dict = tokenizer(`
`992`	`995`	`self.test_data_questions,`
`993`	`996`	`return_tensors="pt",`