From 454b19929a4638d463be9cc70d53a6d71467ecd8 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sat, 8 Mar 2025 03:23:23 -0800 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index f249475edc..ff07ef6917 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -62,7 +62,7 @@ def unsloth_base_fast_generate( if not is_vlm: kwargs["logits_to_keep"] = 1 # Check pad_token - model_eos_token_id = getattr(model.config, "eos_token_id", None) + model_eos_token_id = getattr(self.config, "eos_token_id", None) if model_eos_token_id is not None and hasattr(model_eos_token_id, "__iter__"): model_eos_token_id = model_eos_token_id[0]