From 225c617f48b480a9bb910b3ffcef7e863d0bdb07 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 21 Nov 2024 02:28:00 -0800 Subject: [PATCH] tokenizer_name --- unsloth/models/loader.py | 1 + unsloth/models/vision.py | 1 + 2 files changed, 2 insertions(+) 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: