diff --git a/test/algorithm/test_smooth_quant.py b/test/algorithm/test_smooth_quant.py index 1b8d7a927c4..6bb2f820887 100644 --- a/test/algorithm/test_smooth_quant.py +++ b/test/algorithm/test_smooth_quant.py @@ -737,6 +737,7 @@ def test_sq_qkv(self): sq.transform(alpha=0.5, calib_iter=-1, folding=False) assert isinstance(sq.model.model.decoder.layers[0].self_attn.k_proj, SQLinearWrapper) + class TestExample(unittest.TestCase): def test_sq_quant(self): from neural_compressor import PostTrainingQuantConfig, quantization