Skip to content

Commit 07643fe

Browse files
committed
final fixes
Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags:
1 parent 8376847 commit 07643fe

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchao/quantization/quant_api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -454,7 +454,7 @@ def apply_int4_weight_only_quant(weight):
454454
preserve_zero = False
455455
zero_point_dtype = torch.bfloat16
456456
zero_point_domain = ZeroPointDomain.FLOAT
457-
return to_affine_quantized_intx(weight, mapping_type, block_size, target_dtype, quant_min, quant_max, eps, zero_point_dtype=zero_point_dtype, preserve_zero=preserve_zero, zero_point_domain=zero_point_domain, layout_type=layout_type, use_hqq=use_hq)
457+
return to_affine_quantized_intx(weight, mapping_type, block_size, target_dtype, quant_min, quant_max, eps, zero_point_dtype=zero_point_dtype, preserve_zero=preserve_zero, zero_point_domain=zero_point_domain, layout_type=layout_type, use_hqq=use_hqq)
458458

459459
return _get_linear_subclass_inserter(apply_int4_weight_only_quant)
460460

0 commit comments

Comments
 (0)