diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index bc8dca0ab2..d81348e536 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -493,6 +493,7 @@ class FastVisionModel(FastBaseVisionModel): trust_remote_code = trust_remote_code, revision = revision if not is_peft else None, model_types = model_types, + tokenizer_name = tokenizer_name, *args, **kwargs, ) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index a17c90626c..f088797a8f 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -49,6 +49,7 @@ class FastBaseVisionModel: device_map = "sequential", trust_remote_code = False, model_types = None, + tokenizer_name = None, **kwargs, ): if trust_remote_code: