quick fix (#160)

Co-authored-by: Yang Wang <[email protected]>
microsoft · Jan 13, 2025 · 845ff8c · 845ff8c
1 parent 85d5e8e
commit 845ff8c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vptq/utils/pack.py b/vptq/utils/pack.py
@@ -377,7 +377,7 @@ def absorb_perm_layer(layer):
 def absorb_perm(model):
     absorbed_perm = False
     # Process all VQuantLinear layers
-    pbar = tqdm(model.named_modules())
+    pbar = tqdm.tqdm(model.named_modules())
 
     for name, module in pbar:
         if isinstance(module, vptq.layers.vqlinear.VQuantLinear):