Skip to content

GitLab

Unverified Commit f42d9dcc authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Patch T5 device test (#12742)

parent 370be9cc
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_modeling_t5.py
View file @ f42d9dcc
...@@ -802,7 +802,7 @@ class T5ModelIntegrationTests(unittest.TestCase): ...@@ -802,7 +802,7 @@ class T5ModelIntegrationTests(unittest.TestCase):
model.config.do_sample = False model.config.do_sample = False
tokenizer = T5Tokenizer.from_pretrained("t5-small") tokenizer = T5Tokenizer.from_pretrained("t5-small")
input_ids = tokenizer("summarize: Hello there", return_tensors="pt").input_ids input_ids = tokenizer("summarize: Hello there", return_tensors="pt").input_ids.to(torch_device)
sequences = model.generate(input_ids) sequences = model.generate(input_ids)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment