From 75e1d3537f579a7403658ab64221ffb90f8ba2b9 Mon Sep 17 00:00:00 2001 From: bursteratom Date: Fri, 6 Dec 2024 14:54:06 -0500 Subject: [PATCH] qwen2_vl get_text_config --- src/axolotl/utils/models.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/axolotl/utils/models.py b/src/axolotl/utils/models.py index 884dd0423..c354b5325 100644 --- a/src/axolotl/utils/models.py +++ b/src/axolotl/utils/models.py @@ -372,7 +372,11 @@ class ModelLoader: # init model config self.model_config = load_model_config(cfg) if cfg.is_multimodal: - self.text_model_config = self.model_config.text_config + try: + self.text_model_config = self.model_config.text_config + except AttributeError: + # for qwen2_vl + self.text_model_config = self.model_config.get_text_config() else: self.text_model_config = self.model_config