From a3003024864d45f08953bc4b32d990fbd40d504e Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Sat, 9 May 2026 14:57:28 +0000
Subject: [PATCH] Fix: use hf_ptq.py arg names (--pyt_ckpt_path, --qformat,
 --inference_tensor_parallel)

---
 scripts/quantize_nvfp4.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/scripts/quantize_nvfp4.py b/scripts/quantize_nvfp4.py
index f45d11d..4532a73 100644
--- a/scripts/quantize_nvfp4.py
+++ b/scripts/quantize_nvfp4.py
@@ -212,12 +212,12 @@ def run_calibration(model_path, export_dir, calib_save_path, amax_snapshot_path,
     saved_argv = sys.argv
     sys.argv = [
         "hf_ptq.py",
-        "--model", model_path,
-        "--quant", QUANT,
+        "--pyt_ckpt_path", model_path,
+        "--qformat", QUANT,
         "--calib_size", str(calib_size),
         "--calib_seq", str(calib_seq),
         "--kv_cache_qformat", KV_CACHE_QUANT,
-        "--tp", str(TP),
+        "--inference_tensor_parallel", str(TP),
         "--export_path", export_dir,
         "--trust_remote_code",
         "--use_seq_device_map",