diff --git a/gptqmodel/utils/model.py b/gptqmodel/utils/model.py index 68ccf0148..ec4290ead 100644 --- a/gptqmodel/utils/model.py +++ b/gptqmodel/utils/model.py @@ -516,6 +516,7 @@ def pack_module(name, qModules, quant_result, layers, pbar=None): qModules[name].pack(linear=layers[name], scales=scale, zeros=zero, g_idx=g_idx) qModules[name].to(layer_device) if pbar: + pbar.iter() pbar.progress()