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(
260
260
input = torch .eye (input_size , dtype = torch .bfloat16 )
261
261
output = linear (input )
262
262
263
- print (output .T )
264
- print (torch .nn .functional .mse_loss (output .T , linear .weight ))
265
263
assert torch .allclose (output .T , exp_quant .to (output .dtype ))
266
264
assert torch .nn .functional .mse_loss (output .T , linear .weight ) <= exp_loss
267
265
@@ -376,8 +374,5 @@ def test_static_activation_quantization(
376
374
assert torch .equal (observer .max_val [key ], exp_max_val [key ])
377
375
378
376
# check forward pass
379
- print (args .strategy )
380
- print (output )
381
- print (torch .nn .functional .mse_loss (output , input ))
382
377
assert torch .allclose (output , exp_quant .to (output .dtype ))
383
378
assert torch .nn .functional .mse_loss (output , input ) <= exp_loss
You can’t perform that action at this time.
0 commit comments