diff --git a/examples/diffusers/quantization/quantize.py b/examples/diffusers/quantization/quantize.py index 1d71c088652..06330c56d0d 100644 --- a/examples/diffusers/quantization/quantize.py +++ b/examples/diffusers/quantization/quantize.py @@ -563,7 +563,16 @@ def create_argument_parser() -> argparse.ArgumentParser: calib_group = parser.add_argument_group("Calibration Configuration") calib_group.add_argument("--batch-size", type=int, default=2, help="Batch size for calibration") calib_group.add_argument( - "--calib-size", type=int, default=128, help="Total number of calibration samples" + "--calib-size", + "--calib-samples", + dest="calib_size", + type=int, + default=128, + help=( + "Number of calibration samples/prompts to run (not denoising steps). " + "Total prompts used ~ ceil(calib_size / batch_size) * batch_size. " + "Alias: --calib-samples. For denoising steps, use --n-steps." + ), ) calib_group.add_argument("--n-steps", type=int, default=30, help="Number of denoising steps") calib_group.add_argument(