Make more test models smaller (#25005)

* Make more test models tiny * Make more test models tiny * More models * More models
2023-07-24 10:08:47 -04:00
parent 8f1f0bf50f
commit 42571f6eb8
22 changed files with 149 additions and 137 deletions
--- a/tests/models/esm/test_modeling_esm.py
+++ b/tests/models/esm/test_modeling_esm.py
@@ -279,10 +279,6 @@ class EsmModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
    def test_resize_tokens_embeddings(self):
        pass

-    @unittest.skip("Will be fixed soon by reducing the size of the model used for common tests.")
-    def test_model_is_small(self):
-        pass
-

@require_torch
 class EsmModelIntegrationTest(TestCasePlus):