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

* Fix last models for common tests that are too big. * Remove print statement
2023-07-25 07:56:04 -04:00
parent ee1eb3b325
commit f295fc8a16
19 changed files with 109 additions and 108 deletions
--- a/tests/test_modeling_common.py
+++ b/tests/test_modeling_common.py
@@ -2708,7 +2708,6 @@ 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)

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