From 2af3c9c4daf1a1e574815fff252ff021a4db98c3 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Fri, 19 Sep 2025 04:01:52 -0700 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 24 ++++++++++++++++++------ 1 file changed, 18 insertions(+), 6 deletions(-) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 1e54ac04e2..3f5ae816ea 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -43,7 +43,12 @@ from transformers.models.llama.modeling_llama import logger from transformers import __version__ as transformers_version from triton import __version__ as triton_version from unsloth_zoo.utils import _get_dtype -from unsloth_zoo.hf_utils import dtype_from_config, add_dtype_kwargs, fix_lora_auto_mapping +from unsloth_zoo.hf_utils import ( + dtype_from_config, + add_dtype_kwargs, + fix_lora_auto_mapping, + get_auto_processor, +) from unsloth_zoo.patching_utils import patch_model_and_tokenizer from unsloth_zoo.training_utils import prepare_model_for_training @@ -574,11 +579,18 @@ class FastBaseModel: task = whisper_task, ) else: - tokenizer = auto_processor.from_pretrained( - tokenizer_name, - padding_side = "right", - token = token, - ) + try: + tokenizer = auto_processor.from_pretrained( + tokenizer_name, + padding_side = "right", + token = token, + ) + except: + tokenizer = get_auto_processor( + tokenizer_name, + padding_side = "right", + token = token, + ) if hasattr(tokenizer, "tokenizer"): __tokenizer = tokenizer.tokenizer # Add padding side as well