Skip to content

Commit 5634a89

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 4e1f892 commit 5634a89

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

test/3x/onnxrt/quantization/layer_wise/test_layer_wise.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -149,13 +149,12 @@ def test_gptq_layer_wise(self):
149149
self.assertIsNotNone(lwq_quantized_weight)
150150
quantized_weight, quantized_scale = self._get_quantized_matmul_weight(qmodel, "/lm_head/MatMul_Q4")
151151
self.assertIsNotNone(quantized_weight)
152-
print('lwq_quantized_weight', lwq_quantized_weight)
153-
print('lwq_quantized_scale', lwq_quantized_scale)
154-
print('quantized_weight', quantized_weight)
155-
print('quantized_scale', quantized_scale)
152+
print("lwq_quantized_weight", lwq_quantized_weight)
153+
print("lwq_quantized_scale", lwq_quantized_scale)
154+
print("quantized_weight", quantized_weight)
155+
print("quantized_scale", quantized_scale)
156156
self.assertTrue((lwq_quantized_weight == quantized_weight).all())
157157

158158

159-
160159
if __name__ == "__main__":
161160
unittest.main()

0 commit comments

Comments
 (0)