Skip to content

GitLab

"tests/vscode:/vscode.git/clone" did not exist on "0f38c8597fa77534fbafdac030e578495a2c6f3e"
Commit 66c82765 authored by patrickvonplaten's avatar patrickvonplaten
Browse files

fix typo in test gpt2

parent 314bdc7c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_modeling_gpt2.py
View file @ 66c82765
...@@ -343,7 +343,7 @@ class GPT2ModelLanguageGenerationTest(unittest.TestCase): ...@@ -343,7 +343,7 @@ class GPT2ModelLanguageGenerationTest(unittest.TestCase):
@slow @slow
def test_lm_generate_gpt2(self): def test_lm_generate_gpt2(self):
model = GPT2LMHeadModel.from_pretrained("gpt2") 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 = [ expected_output_ids = [
464, 464,
3290, 3290,
......
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