From f42d9dcc0e07b990d76c49e7af9ac927e50a4546 Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Thu, 15 Jul 2021 17:40:17 +0200 Subject: [PATCH] Patch T5 device test (#12742) --- tests/test_modeling_t5.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_t5.py b/tests/test_modeling_t5.py index 659cf9b0c7..38a353e1a2 100644 --- a/tests/test_modeling_t5.py +++ b/tests/test_modeling_t5.py @@ -802,7 +802,7 @@ class T5ModelIntegrationTests(unittest.TestCase): model.config.do_sample = False 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)