Fix: use full argparse flag names (--calib_size, --kv_cache_qformat)
This commit is contained in:
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user