Skip to content

Commit 1104f44

Browse files
committed
fix bug
Signed-off-by: xin3he <xin3.he@intel.com>
1 parent b8429a5 commit 1104f44

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/3x/torch/quantization/weight_only/test_mixed_algos.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@ def run_fn(model):
1818
class TestMixedTwoAlgo:
1919
def test_mixed_gptq_and_rtn(self):
2020
with patch.object(logger, "info") as mock_info:
21-
rtn_config = RTNConfig(quant_lm_head=True)
22-
gptq_config = GPTQConfig(double_quant_bits=4, white_list=["transformer.*"])
21+
rtn_config = RTNConfig(white_list=[".*mlp.*"])
22+
gptq_config = GPTQConfig(double_quant_bits=4, white_list=[".*attn.*"])
2323
combined_config = rtn_config + gptq_config
2424
logger.info(combined_config)
2525

0 commit comments

Comments
 (0)