From 355cd7c872cef4b56d3108db21511f97d36aea0d Mon Sep 17 00:00:00 2001 From: bursteratom Date: Fri, 6 Dec 2024 14:43:50 -0500 Subject: [PATCH] update is_multimodal requirement to include qwen2_vl --- src/axolotl/utils/config/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/utils/config/__init__.py b/src/axolotl/utils/config/__init__.py index 468bd6e7f..3737b00ea 100644 --- 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 [