update is_multimodal requirement to include qwen2_vl

2024-12-06 14:43:50 -05:00
parent eab1638686
commit 355cd7c872
1 changed files with 1 additions and 1 deletions
--- a/src/axolotl/utils/config/init.py
+++ b/src/axolotl/utils/config/init.py
@@ -132,7 +132,7 @@ def normalize_config(cfg):

    cfg.is_multimodal = (
        hasattr(model_config, "model_type")
-        and model_config.model_type in ["llava", "mllama"]
+        and model_config.model_type in ["llava", "mllama", "qwen2_vl"]
        or any(
            multimodal_name in cfg.base_model.lower()
            for multimodal_name in [