add Mllama/Llava class

Kaihui-intel · Kaihui-intel · commit 406f93b3740d · 2025-02-10T08:56:38.000+08:00
Signed-off-by: Kaihui-intel &lt;kaihui.tang@intel.com&gt;
diff --git a/neural_compressor/transformers/models/__init__.py b/neural_compressor/transformers/models/__init__.py
@@ -13,4 +13,11 @@
 # limitations under the License.
 
 from .modeling_auto import _BaseINCAutoModelClass
-from .modeling_auto import AutoModel, AutoModelForCausalLM, AutoModelForSeq2SeqLM, Qwen2VLForConditionalGeneration
+from .modeling_auto import (
+    AutoModel,
+    AutoModelForCausalLM,
+    AutoModelForSeq2SeqLM,
+    Qwen2VLForConditionalGeneration,
+    MllamaForConditionalGeneration, 
+    LlavaForConditionalGeneration,
+)
