update

ModelTC · Jun 18, 2024 · c9534f9 · c9534f9
1 parent 32a1c46
commit c9534f9
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/llmc/models/llava.py b/llmc/models/llava.py
@@ -1,6 +1,14 @@
+from loguru import logger
 from .llama import Llama
 from llmc.utils.registry_factory import MODEL_REGISTRY
-from transformers import LlavaForConditionalGeneration, AutoConfig
+from transformers import AutoConfig
+
+try:
+ from transformers import LlavaForConditionalGeneration
+except:
+ logger.info(
+ "LlavaForConditionalGeneration is not supported in this version of transfomers. Update transfomers if you need."
+ )
 
 
 @MODEL_REGISTRY