From 314bdc7c14452862e4fa498e84a07c9a950b8980 Mon Sep 17 00:00:00 2001 From: patrickvonplaten Date: Sun, 8 Mar 2020 15:34:20 +0100 Subject: [PATCH] fix typo in test --- 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 132bc9e621..7c85c7290b 100644 --- a/tests/test_modeling_gpt2.py +++ b/tests/test_modeling_gpt2.py @@ -372,7 +372,7 @@ class GPT2ModelLanguageGenerationTest(unittest.TestCase): @slow def test_lm_generate_distilgpt2(self): model = GPT2LMHeadModel.from_pretrained("distilgpt2") - input_ids = torch.tensor([[463, 1893]], dtype=torch.long, device=torch_device) # The president + input_ids = torch.tensor([[464, 1893]], dtype=torch.long, device=torch_device) # The president expected_output_ids = [ 464, 1893,