From 407f5d472bf19083a97d50eca06dd8bfd51998a6 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 11 Mar 2025 20:43:25 -0700 Subject: [PATCH] Update loader.py --- unsloth/models/loader.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index 876deaec53..f359861060 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -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)