We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 32f8ec9 commit 03797cdCopy full SHA for 03797cd
modelopt/torch/quantization/model_quant.py
@@ -107,7 +107,7 @@ def forward_loop(model):
107
108
for name, module in model.named_modules():
109
if isinstance(module, TensorQuantizer):
110
- for attr_name in ["_amax", "_bias", "_pre_quant_scale"]:
+ for attr_name in ["_amax", "_bias_value", "_pre_quant_scale"]:
111
module.validate_attr(attr_name=attr_name, warn_error=True, name=name)
112
113
# TODO: Re-enable when the CUDA error: unspecified launch failure is fixed.
0 commit comments