From 366947bb2576fa2f0180096f35f8b8a2d1dfa1d3 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]