From f9a7748bd80291b53f72920586875666fbc4bdb4 Mon Sep 17 00:00:00 2001 From: Chirag Jain Date: Sat, 7 Dec 2024 15:32:32 +0530 Subject: [PATCH] Fix llama type model check (#2142) [skip ci] --- 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 468bd6e7f8..30ba53ad25 100644 --- a/src/axolotl/utils/config/__init__.py +++ b/src/axolotl/utils/config/__init__.py @@ -153,7 +153,7 @@ def normalize_config(cfg): cfg.is_llama_derived_model = ( ( hasattr(model_config, "model_type") - and model_config.model_type == ["llama", "mllama_text_model"] + and model_config.model_type in ["llama", "mllama_text_model"] ) or cfg.is_llama_derived_model or "llama" in cfg.base_model.lower()