ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update loader.py

Browse source
This commit is contained in:
Daniel Han 2025-03-11 20:43:25 -07:00
commit 407f5d472b
1 changed files with 12 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

12
unsloth/models/loader.py
View file

@ -487,6 +487,18 @@ class FastModel(FastBaseModel):
if not use_exact_model_name:
model_name = get_model_name(model_name, load_in_4bit)
# Check versions
if "pixtral" in model_name.lower() and transformers_version < Version("4.49.0"):
raise RuntimeError(
"Unsloth: Pixtral only works on transformers >= 4.49.0."\
"Please update transformers via `pip install --upgrade transformers>=4.49.0`"
)
elif "qwen2.5" in model_name.lower() and transformers_version < Version("4.49.0"):
raise RuntimeError(
"Unsloth: Qwen 2.5 only works on transformers >= 4.49.0."\
"Please update transformers via `pip install --upgrade transformers>=4.49.0`"
)
if USE_MODELSCOPE and not os.path.exists(model_name):
from modelscope import snapshot_download
model_name = snapshot_download(model_name)