"...git@developer.sourcefind.cn:modelzoo/solov2-pytorch.git" did not exist on "b1ba5939d6128335f77b87da1ef747cf06c8e114"
Unverified Commit 52decab3 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

fix test (#7947)

parent 9b6610f7
......@@ -446,7 +446,6 @@ class GPT2ModelTest(ModelTesterMixin, unittest.TestCase):
inputs = tokenizer(sentences, return_tensors="pt", padding=True)
torch.manual_seed(0)
outputs = model.generate(
input_ids=inputs["input_ids"].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