Skip to content

Commit 339dc00

Browse files
fix progress bar for packing
1 parent c3ea803 commit 339dc00

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

gptqmodel/utils/model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -516,6 +516,7 @@ def pack_module(name, qModules, quant_result, layers, pbar=None):
516516
qModules[name].pack(linear=layers[name], scales=scale, zeros=zero, g_idx=g_idx)
517517
qModules[name].to(layer_device)
518518
if pbar:
519+
pbar.current_iteration += 1
519520
pbar.progress()
520521

521522

0 commit comments

Comments
 (0)