Edit err message and comment in test_model_is_small (#25087)

* Edit err message and comment in * put back 80M comment
2023-07-25 12:24:36 -04:00
parent 2fac342238
commit 0779fc8eb8
1 changed files with 1 additions and 1 deletions
--- a/tests/test_modeling_common.py
+++ b/tests/test_modeling_common.py
@@ -2714,7 +2714,7 @@ class ModelTesterMixin:
            num_params = model.num_parameters()
            assert (
                num_params < 1000000
-            ), f"{model_class} is too big for the common tests ({num_params})! It should have 200k max."
+            ), f"{model_class} is too big for the common tests ({num_params})! It should have 1M max."


 global_rng = random.Random()