From 25b4d8da061288fc29a6715dd1d2e35fd201c7ca Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sat, 9 May 2026 13:37:24 +0000 Subject: [PATCH] Fix: add missing args for make_calib_dataloader (dataset, calib_with_images, auto_quantize, specdec) --- scripts/quantize_nvfp4.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/scripts/quantize_nvfp4.py b/scripts/quantize_nvfp4.py index c671f81..1687536 100644 --- a/scripts/quantize_nvfp4.py +++ b/scripts/quantize_nvfp4.py @@ -281,8 +281,14 @@ def run_calibration(model_path, export_dir, calib_save_path, amax_snapshot_path, calib_size=[calib_size], calib_seq=calib_seq, calib_dataset="", + dataset=None, # None triggers default: ["cnn_dailymail", "nemotron-post-training-dataset-v2"] batch_size=batch_size, calib_batch_size=0, + calib_with_images=False, + auto_quantize_bits=None, + auto_quantize_method=None, + specdec_offline_dataset=None, + inference_pipeline_parallel=1, ) calib_dataloader, _ = make_calib_dataloader( args, model, None, tokenizer, torch.device("cuda"), None,