Fix (brevitas_examples/generative): scaling_min_val for any dtype (#1117

)
Xilinx · Dec 6, 2024 · 0ea7bac · 0ea7bac
1 parent 85c1626
commit 0ea7bac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas_examples/common/generative/quantize.py b/src/brevitas_examples/common/generative/quantize.py
@@ -318,7 +318,7 @@ def generate_quantizers(
     if weight_group_dim is not None:
         weight_quant = weight_quant.let(**{'group_dim': weight_group_dim})
 
-    if dtype != torch.float32:
+    if scaling_min_val is not None:
         weight_quant = weight_quant.let(**{'scaling_min_val': scaling_min_val})
         input_quant = input_quant.let(
             **{'scaling_min_val': scaling_min_val}) if input_quant is not None else None