Skip to content
2 changes: 1 addition & 1 deletion unsloth/models/_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -251,7 +251,7 @@ def patch_mistral_nemo_config(config):

from transformers import __version__ as transformers_version
from transformers import PretrainedConfig
model_architectures = ["llama", "mistral", "gemma", "gemma2", "qwen2", "granite", "qwen3", "qwen3_moe"]
model_architectures = ["llama", "mistral", "gemma", "gemma2", "qwen2", "granite", "qwen3", "qwen3_moe", "falcon_h1"]

for model_name in model_architectures:
config_filepath = f"transformers.models.{model_name}.configuration_{model_name}"
Expand Down
Loading