Skip to content

Commit 4e59790

Browse files
committed
updated stale comment
Signed-off-by: Suguna Velury <178320438+sugunav14@users.noreply.github.com>
1 parent fc6ba07 commit 4e59790

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

modelopt/torch/quantization/model_calib.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1861,7 +1861,7 @@ def _layer_forward_loop(m):
18611861
for args, kwargs_input in inputs: # noqa: F821
18621862
m(*args, **kwargs_input)
18631863

1864-
# Call GPTQ
1865-
calib_func(layer, inputs, forward_loop=_layer_forward_loop, **calib_kwargs)
1864+
# Call calibration function
1865+
calib_func(layer, _layer_forward_loop, **calib_kwargs)
18661866
del inputs
18671867
torch.cuda.empty_cache()

0 commit comments

Comments
 (0)