Skip to content

Commit d313cfa

Browse files
authored
fix syntax error in eval_combo.py (#852)
Summary: Fixing invalid syntax, this is breaking internal tests. We should follow up with an automated way to catch this in OSS, but unbreaking first. Test Plan: CI Reviewers: Subscribers: Tasks: Tags:
1 parent e1039ab commit d313cfa

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchao/_models/sam/eval_combo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -331,7 +331,7 @@ def mlp_only(mod, name):
331331
int8_dynamic_activation_int8_weight(),
332332
attn_only)
333333
quantize_(predictor.model.image_encoder, int4_weight_only(layout_type=MarlinSparseLayoutType()), mlp_lin1_only)
334-
sparsify_(predictor.model.image_encoder
334+
sparsify_(predictor.model.image_encoder,
335335
semi_sparse_weight(),
336336
mlp_lin2_only)
337337
if not TORCH_VERSION_AT_LEAST_2_5:

0 commit comments

Comments
 (0)