File tree Expand file tree Collapse file tree 1 file changed +0
-5
lines changed
tests/llmcompressor/modifiers/calibration Expand file tree Collapse file tree 1 file changed +0
-5
lines changed Original file line number Diff line number Diff line change @@ -260,8 +260,6 @@ def test_static_weight_quantization(
260260 input = torch .eye (input_size , dtype = torch .bfloat16 )
261261 output = linear (input )
262262
263- print (output .T )
264- print (torch .nn .functional .mse_loss (output .T , linear .weight ))
265263 assert torch .allclose (output .T , exp_quant .to (output .dtype ))
266264 assert torch .nn .functional .mse_loss (output .T , linear .weight ) <= exp_loss
267265
@@ -376,8 +374,5 @@ def test_static_activation_quantization(
376374 assert torch .equal (observer .max_val [key ], exp_max_val [key ])
377375
378376 # check forward pass
379- print (args .strategy )
380- print (output )
381- print (torch .nn .functional .mse_loss (output , input ))
382377 assert torch .allclose (output , exp_quant .to (output .dtype ))
383378 assert torch .nn .functional .mse_loss (output , input ) <= exp_loss
You can’t perform that action at this time.
0 commit comments