From da46258d222e37630c3820ef2f7f7d43a372d0f0 Mon Sep 17 00:00:00 2001 From: n1ck-guo Date: Tue, 4 Nov 2025 19:33:48 -0500 Subject: [PATCH 1/2] fix bug of cannot create adam compressor Signed-off-by: n1ck-guo --- auto_round/compressors/__init__.py | 2 +- auto_round/compressors/base.py | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/auto_round/compressors/__init__.py b/auto_round/compressors/__init__.py index 03983b2c7..6f8ddf681 100644 --- a/auto_round/compressors/__init__.py +++ b/auto_round/compressors/__init__.py @@ -14,7 +14,7 @@ from auto_round.compressors.adam import AdamCompressor from auto_round.compressors.base import BaseCompressor -from auto_round.compressors.base import BaseCompressor as LLMCompressor +from auto_round.compressors.base import LLMCompressor from auto_round.compressors.mllm.compressor import MLLMCompressor from auto_round.compressors.diffusion.compressor import DiffusionCompressor from auto_round.compressors.config import ( diff --git a/auto_round/compressors/base.py b/auto_round/compressors/base.py index 5e00f92a4..fa0c4c7e0 100644 --- a/auto_round/compressors/base.py +++ b/auto_round/compressors/base.py @@ -3107,3 +3107,7 @@ def _sampling_inputs( current_input_others[key] = input_others[key] return current_input_ids, current_input_others + + +class LLMCompressor(BaseCompressor): + pass From f68514ed4e39127afbdebf6836203ae267fd742f Mon Sep 17 00:00:00 2001 From: n1ck-guo Date: Tue, 4 Nov 2025 19:51:00 -0500 Subject: [PATCH 2/2] add ut Signed-off-by: n1ck-guo --- test/test_cpu/test_autoround.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/test/test_cpu/test_autoround.py b/test/test_cpu/test_autoround.py index 626dec380..980db236d 100644 --- a/test/test_cpu/test_autoround.py +++ b/test/test_cpu/test_autoround.py @@ -796,6 +796,12 @@ def test_attention_mask_via_tokenize_in_dataset(self): ar = AutoRound(model_name, iters=1, dataset=data, seqlen=8) ar.quantize() + def test_create_adam(self): + model_name = "/tf_dataset/auto_round/models/Qwen/Qwen3-0.6B" + from auto_round import AutoRound + + ar = AutoRound(model=model_name, enable_adam=True) + if __name__ == "__main__": unittest.main()