Update ruff to 0.11.2 (#36962)
* update * update * update --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
@@ -3234,9 +3234,9 @@ class ModelTesterMixin:
|
||||
for model_class in self.all_model_classes:
|
||||
model = model_class(config)
|
||||
num_params = model.num_parameters()
|
||||
assert (
|
||||
num_params < 1000000
|
||||
), f"{model_class} is too big for the common tests ({num_params})! It should have 1M max."
|
||||
assert num_params < 1000000, (
|
||||
f"{model_class} is too big for the common tests ({num_params})! It should have 1M max."
|
||||
)
|
||||
|
||||
@require_flash_attn
|
||||
@require_torch_gpu
|
||||
|
||||
Reference in New Issue
Block a user