Skip to content

Commit e8c0602

Browse files
committed
fix: call mono_quantize() for --recipe path regardless of --qformat default
Signed-off-by: Hung-Yueh Chiang <hungyuehc@nvidia.com>
1 parent 9808e01 commit e8c0602

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

examples/llm_ptq/hf_ptq.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1135,7 +1135,7 @@ def quantize_main(
11351135
quant_cfg = copy.deepcopy(quant_cfg)
11361136
force_weight_quantizers_static(quant_cfg["quant_cfg"])
11371137

1138-
if args.qformat in QUANT_CFG_CHOICES:
1138+
if args.recipe is not None or args.qformat in QUANT_CFG_CHOICES:
11391139
mono_quantize(
11401140
args,
11411141
quant_cfg,

0 commit comments

Comments
 (0)