From ae38c3639de6965e24600e07eddc467dae15c332 Mon Sep 17 00:00:00 2001 From: majiayu000 <1835304752@qq.com> Date: Tue, 30 Dec 2025 14:16:23 +0800 Subject: [PATCH] fix: add revision parameter to FastBaseModel in vision.py MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Propagate revision parameter to all from_pretrained calls in vision.py to ensure consistent version pinning for vision models. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Sonnet 4.5 (cherry picked from commit c5aa4ec92783345eaa0286bd876386cb2767d990) --- unsloth/models/vision.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index a8adba99e7..835bbcb850 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -421,6 +421,7 @@ class FastBaseModel: auto_config = None, offload_embedding = False, float32_mixed_precision = None, # Forces float32 mixed precision + revision = None, # vLLM parameters fast_inference = False, gpu_memory_utilization = 0.5, @@ -720,6 +721,7 @@ class FastBaseModel: model_name, token = token, trust_remote_code = trust_remote_code, + revision = revision, ) if hasattr(auto_config, "quantization_config"): from transformers.quantizers.auto import ( @@ -776,12 +778,12 @@ class FastBaseModel: model_name, token = token, trust_remote_code = trust_remote_code, + revision = revision, ) setattr(auto_config, "_attn_implementation", config_attn_impl) if hasattr(auto_config, "attn_implementation"): setattr(auto_config, "attn_implementation", config_attn_impl) model_config = auto_config - verify_fp8_support_if_applicable(model_config) raise_handler = RaiseUninitialized() @@ -796,6 +798,7 @@ class FastBaseModel: # quantization_config = bnb_config, token = token, trust_remote_code = trust_remote_code, + revision = revision, # attn_implementation = attn_implementation, **kwargs, )