Skip to content

Commit 4d1c1ff

Browse files
committed
added calib_func check
Signed-off-by: Suguna Velury <178320438+sugunav14@users.noreply.github.com>
1 parent a938963 commit 4d1c1ff

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

modelopt/torch/quantization/mode.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -229,6 +229,9 @@ def wrapped_calib_func(
229229

230230
if func is not None:
231231
if sequential:
232+
assert method in ["max"], (
233+
f"Sequential calibration currently only supports max calibration, got {method}"
234+
)
232235
# Wrap with sequential processing
233236
sequential_calibrate(
234237
model,

0 commit comments

Comments
 (0)