SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Edit err message and comment in test_model_is_small (#25087)

Browse Source 
* Edit err message and comment in

* put back 80M comment
This commit is contained in:
Connor Henderson
2023-07-25 12:24:36 -04:00
committed by GitHub
parent 2fac342238
commit 0779fc8eb8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_modeling_common.py
View File

@@ -2714,7 +2714,7 @@ class ModelTesterMixin:
num_params = model.num_parameters() num_params = model.num_parameters()
assert ( assert (
num_params < 1000000 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() global_rng = random.Random()