diff --git a/neural_compressor/transformers/models/modeling_auto.py b/neural_compressor/transformers/models/modeling_auto.py index 281a19dcf56..efd38a719dc 100644 --- a/neural_compressor/transformers/models/modeling_auto.py +++ b/neural_compressor/transformers/models/modeling_auto.py @@ -758,12 +758,21 @@ class AutoModelForSeq2SeqLM(_BaseINCAutoModelClass): class Qwen2VLForConditionalGeneration(_BaseINCAutoModelClass): - ORIG_MODEL = transformers.Qwen2VLForConditionalGeneration + if transformers.__version__ >= "4.46": + ORIG_MODEL = transformers.Qwen2VLForConditionalGeneration + else: + logger.warning("please install transformers>=4.46 for quantizing Qwen2VLForConditionalGeneration.") class MllamaForConditionalGeneration(_BaseINCAutoModelClass): - ORIG_MODEL = transformers.MllamaForConditionalGeneration + if transformers.__version__ >= "4.46": + ORIG_MODEL = transformers.MllamaForConditionalGeneration + else: + logger.warning("please install transformers>=4.46 for quantizing MllamaForConditionalGeneration.") class LlavaForConditionalGeneration(_BaseINCAutoModelClass): - ORIG_MODEL = transformers.LlavaForConditionalGeneration + if transformers.__version__ >= "4.46": + ORIG_MODEL = transformers.LlavaForConditionalGeneration + else: + logger.warning("please install transformers>=4.46 for quantizing LlavaForConditionalGeneration.")