Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "63e539459d5d241eab8c523be027a9bdfc7df3b4"
Commit 314bdc7c authored by patrickvonplaten's avatar patrickvonplaten
Browse files

fix typo in test

parent 57597614
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_modeling_gpt2.py
View file @ 314bdc7c
...@@ -372,7 +372,7 @@ class GPT2ModelLanguageGenerationTest(unittest.TestCase): ...@@ -372,7 +372,7 @@ class GPT2ModelLanguageGenerationTest(unittest.TestCase):
@slow @slow
def test_lm_generate_distilgpt2(self): def test_lm_generate_distilgpt2(self):
model = GPT2LMHeadModel.from_pretrained("distilgpt2") 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 = [ expected_output_ids = [
464, 464,
1893, 1893,
......
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