From 1a36a655eab184532a053df33b9ca705057bb88f Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sat, 9 May 2026 14:54:51 +0000 Subject: [PATCH] Fix: use full argparse flag names (--calib_size, --kv_cache_qformat) --- scripts/quantize_nvfp4.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/quantize_nvfp4.py b/scripts/quantize_nvfp4.py index c1462ac..f45d11d 100644 --- a/scripts/quantize_nvfp4.py +++ b/scripts/quantize_nvfp4.py @@ -214,9 +214,9 @@ def run_calibration(model_path, export_dir, calib_save_path, amax_snapshot_path, "hf_ptq.py", "--model", model_path, "--quant", QUANT, - "--calib", str(calib_size), + "--calib_size", str(calib_size), "--calib_seq", str(calib_seq), - "--kv_cache_quant", KV_CACHE_QUANT, + "--kv_cache_qformat", KV_CACHE_QUANT, "--tp", str(TP), "--export_path", export_dir, "--trust_remote_code",