Fix validate

Xilinx · Jun 27, 2023 · e6a7d70 · e6a7d70
1 parent 5b6f975
commit e6a7d70
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/brevitas_examples/llm/main.py b/src/brevitas_examples/llm/main.py
@@ -164,13 +164,13 @@ def validate(args):
             assert args.weight_quant_granularity != 'per_group', "ONNX QCDQ export doesn't support group weight quantization."
             if args.weight_quant_type == 'asym':
                 assert args.quantize_weight_zero_point, "Quantized weight zero point required."
-            if args.input_quant_type == 'asym':
+            if args.input_bit_width is not None and args.input_quant_type == 'asym':
                 assert args.quantize_input_zero_point, "Quantized input zero point required."
         if args.export_target == 'torch_qcdq':
             assert args.weight_quant_granularity != 'per_group', "TorchScript QCDQ export doesn't support group weight quantization."
             if args.weight_quant_type == 'asym':
                 assert args.quantize_weight_zero_point, "Quantized weight zero point required."
-            if args.input_quant_type == 'asym':
+            if args.input_bit_width is not None and args.input_quant_type == 'asym':
                 assert args.quantize_input_zero_point, "Quantized input zero point required."
         if args.input_bit_width:
             assert args.act_calibration, "Input quantization is being applied without activation calibration. Set --act-calibration."