Skip to content

Commit 03797cd

Browse files
committed
minor
1 parent 32f8ec9 commit 03797cd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modelopt/torch/quantization/model_quant.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ def forward_loop(model):
107107

108108
for name, module in model.named_modules():
109109
if isinstance(module, TensorQuantizer):
110-
for attr_name in ["_amax", "_bias", "_pre_quant_scale"]:
110+
for attr_name in ["_amax", "_bias_value", "_pre_quant_scale"]:
111111
module.validate_attr(attr_name=attr_name, warn_error=True, name=name)
112112

113113
# TODO: Re-enable when the CUDA error: unspecified launch failure is fixed.

0 commit comments

Comments
 (0)