riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-17 19:48:23 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

[Pix2Struct] Fix slow test (#22448)

Browse Source 
fix slow test
This commit is contained in:
Younes Belkada 2023-03-29 17:40:45 +02:00 committed by GitHub
parent 55dae94c0c
commit b844f8a9ab
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/pix2struct/test_modeling_pix2struct.py
View File

@ -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(