Fix (ptq): disable MHA input_quant in graph quant

Xilinx · Nov 28, 2023 · 4aa9968 · 4aa9968
1 parent 63370f4
commit 4aa9968
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py b/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py
@@ -345,6 +345,7 @@ def kwargs_prefix(prefix, weight_kwargs):
     quant_mha_kwargs = {
         **kwargs_prefix('in_proj_', weight_quant_dict),
         **kwargs_prefix('out_proj_', weight_quant_dict),
+        'in_proj_input_quant': None,
         'in_proj_bias_quant': bias_quant,
         'softmax_input_quant': None,
         'attn_output_weights_quant': sym_act_quant,