Skip to content

GitLab

Unverified Commit b844f8a9 authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

[`Pix2Struct`] Fix slow test (#22448) 

fix slow test
parent 55dae94c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/models/pix2struct/test_modeling_pix2struct.py
View file @ b844f8a9
...@@ -406,7 +406,7 @@ class Pix2StructTextImageModelTest(ModelTesterMixin, unittest.TestCase): ...@@ -406,7 +406,7 @@ class Pix2StructTextImageModelTest(ModelTesterMixin, unittest.TestCase):
def test_model(self): def test_model(self):
config, input_dict = self.model_tester.prepare_config_and_inputs_for_common() config, input_dict = self.model_tester.prepare_config_and_inputs_for_common()
for model_class in self.all_model_classes: for model_class in self.all_model_classes:
model = model_class(config) model = model_class(config).to(torch_device)
output = model(**input_dict) output = model(**input_dict)
self.assertEqual( self.assertEqual(
......
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