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 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()