tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Fix llama type model check (#2142) [skip ci]

Browse Source
This commit is contained in:
Chirag Jain
2024-12-07 15:32:32 +05:30
committed by GitHub
parent 5e9fa33f3d
commit f9a7748bd8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/utils/config/__init__.py
View File

@@ -153,7 +153,7 @@ def normalize_config(cfg):
cfg.is_llama_derived_model = ( cfg.is_llama_derived_model = (
( (
hasattr(model_config, "model_type") 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 cfg.is_llama_derived_model
or "llama" in cfg.base_model.lower() or "llama" in cfg.base_model.lower()