Skip to content

GitLab

"git@developer.sourcefind.cn:wuxk1/megatron-lm.git" did not exist on "01a82723cad4e061c9f5a7bebcc443ad7422b945"
Unverified Commit 52decab3 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

fix test (#7947)

parent 9b6610f7
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
tests/test_modeling_gpt2.py
View file @ 52decab3
...@@ -446,7 +446,6 @@ class GPT2ModelTest(ModelTesterMixin, unittest.TestCase): ...@@ -446,7 +446,6 @@ class GPT2ModelTest(ModelTesterMixin, unittest.TestCase):
inputs = tokenizer(sentences, return_tensors="pt", padding=True) inputs = tokenizer(sentences, return_tensors="pt", padding=True)
torch.manual_seed(0)
outputs = model.generate( outputs = model.generate(
input_ids=inputs["input_ids"].to(torch_device), input_ids=inputs["input_ids"].to(torch_device),
attention_mask=inputs["attention_mask"].to(torch_device), attention_mask=inputs["attention_mask"].to(torch_device),
......
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