update format script

microsoft · Oct 28, 2024 · 8e3c9ec · 8e3c9ec
1 parent 517ecb8
commit 8e3c9ec
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/format.sh b/format.sh
@@ -2,7 +2,11 @@
 
 # Format Python files using yapf
 echo "Running yapf..."
-yapf --recursive . --in-place
+find . -type f -name "*.py" \
+    ! -path "./build/*" \
+    ! -path "./.git/*" \
+    ! -path "*.egg-info/*" \
+    -print0 | xargs -0 yapf --in-place
 
 # Format Python imports using isort
 echo "Running isort..."

diff --git a/vptq/layers/vqlinear.py b/vptq/layers/vqlinear.py
@@ -640,9 +640,7 @@ def dequant(self):
                 if self.vector_quant_dim == "in":
                     assert True, "Not implemented"
                 else:
-                    qweight_outlier = qweight_outlier[
-                        :-self.outlier_padding,
-                    ]
+                    qweight_outlier = qweight_outlier[:-self.outlier_padding,]
             # print('qweight: ', qweight.shape)
             # print('qweight_outlier: ', qweight_outlier.shape)
             qweight = torch.cat([qweight_outlier, qweight], dim=1)