Skip to content

Commit

Permalink
Fix llama type model check (#2142) [skip ci]
Browse files Browse the repository at this point in the history
  • Loading branch information
chiragjn authored Dec 7, 2024
1 parent 5e9fa33 commit f9a7748
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/axolotl/utils/config/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()
Expand Down

0 comments on commit f9a7748

Please sign in to comment.