From 66c827656f1a43d1427961eedf4786c4440c89ee Mon Sep 17 00:00:00 2001 From: patrickvonplaten Date: Sun, 8 Mar 2020 15:35:08 +0100 Subject: [PATCH] fix typo in test gpt2 --- tests/test_modeling_gpt2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_gpt2.py b/tests/test_modeling_gpt2.py index 7c85c7290b..bd5fe6f32c 100644 --- a/tests/test_modeling_gpt2.py +++ b/tests/test_modeling_gpt2.py @@ -343,7 +343,7 @@ class GPT2ModelLanguageGenerationTest(unittest.TestCase): @slow def test_lm_generate_gpt2(self): model = GPT2LMHeadModel.from_pretrained("gpt2") - input_ids = torch.tensor([[463, 3290]], dtype=torch.long, device=torch_device) # The dog + input_ids = torch.tensor([[464, 3290]], dtype=torch.long, device=torch_device) # The dog expected_output_ids = [ 464, 3290,