Update quantize.py

Xilinx · Jan 21, 2025 · 0714261 · 0714261
1 parent a0e9f03
commit 0714261
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
@@ -396,7 +396,7 @@ def generate_quantizers(
                 k_transposed_quant = sym_input_quant.let(
                     **{
                         'group_dim': -2, 'group_size': input_group_size})
-            v_quant = q_scaled_quant
+            v_quant = k_transposed_quant
             attn_output_weights_quant = q_scaled_quant
         else:
             q_scaled_quant = v_quant = k_transposed_quant = attn_output_weights_quant = sym_input_quant