From c2ed0d8a99f920c808b3c3e7950690b6b2d2e0ec Mon Sep 17 00:00:00 2001 From: DoubleMathew Date: Thu, 17 Jul 2025 17:34:14 -0500 Subject: [PATCH] use fastmodel (#2987) --- unsloth/models/loader.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index 683f2b1872..953991d41d 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -320,15 +320,15 @@ class FastLanguageModel(FastLlamaModel): f"to obtain the latest transformers build, then restart this session."\ ) dispatch_model = FastQwen3Model if model_type == "qwen3" else FastQwen3MoeModel - elif model_type == "falcon_h1": - dispatch_model = FastFalconH1Model - if not SUPPORTS_FALCON_H1: - raise ImportError( - f"Unsloth: Your transformers version of {transformers_version} does not support FalconH1.\n"\ - f"The minimum required version is 4.50.3.\n"\ - f'Try `pip install --upgrade "transformers>=4.50.3"`\n'\ - f"to obtain the latest transformers build, then restart this session."\ - ) + # elif model_type == "falcon_h1": + # dispatch_model = FastFalconH1Model + # if not SUPPORTS_FALCON_H1: + # raise ImportError( + # f"Unsloth: Your transformers version of {transformers_version} does not support FalconH1.\n"\ + # f"The minimum required version is 4.50.3.\n"\ + # f'Try `pip install --upgrade "transformers>=4.50.3"`\n'\ + # f"to obtain the latest transformers build, then restart this session."\ + # ) # Temporary disable optimized Cohere until errors match # elif model_type == "cohere": # dispatch_model = FastCohereModel