File tree Expand file tree Collapse file tree 1 file changed +4
-5
lines changed
test/3x/onnxrt/quantization/layer_wise Expand file tree Collapse file tree 1 file changed +4
-5
lines changed Original file line number Diff line number Diff 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-
160159if __name__ == "__main__" :
161160 unittest .main ()
You can’t perform that action at this time.
0 commit comments