Fix last models for common tests that are too big. (#25058)

* Fix last models for common tests that are too big.

* Remove print statement
This commit is contained in:
Sylvain Gugger
2023-07-25 07:56:04 -04:00
committed by GitHub
parent ee1eb3b325
commit f295fc8a16
19 changed files with 109 additions and 108 deletions

View File

@@ -279,10 +279,6 @@ class LayoutLMModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase
config_and_inputs = self.model_tester.prepare_config_and_inputs()
self.model_tester.create_and_check_for_question_answering(*config_and_inputs)
@unittest.skip("Will be fixed soon by reducing the size of the model used for common tests.")
def test_model_is_small(self):
pass
def prepare_layoutlm_batch_inputs():
# Here we prepare a batch of 2 sequences to test a LayoutLM forward pass on: