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/test_modeling_common.py
+++ b/tests/test_modeling_common.py
@@ -2708,7 +2708,7 @@ class ModelTesterMixin:
    def test_model_is_small(self):
        # Just a consistency check to make sure we are not running tests on 80M parameter models.
        config, _ = self.model_tester.prepare_config_and_inputs_for_common()
-        # print(config)
+        print(config)

        for model_class in self.all_model_classes:
            model = model_class(config)