get_text_config

2024-12-06 14:50:05 -05:00
parent 355cd7c872
commit d85a229afe
1 changed files with 6 additions and 1 deletions
--- a/src/axolotl/utils/config/init.py
+++ b/src/axolotl/utils/config/init.py
@@ -145,7 +145,12 @@ def normalize_config(cfg):
        cfg.processor_config = (
            cfg.processor_config or cfg.base_model_config or cfg.base_model
        )
-        model_config = model_config.text_config
+
+        try:
+            model_config = model_config.text_config
+        except AttributeError:
+            # for qwen2_vl
+            model_config = model_config.get_text_config()

    cfg.model_config_type = model_config.model_type