From a22eec898d6539e8b50ae1d99009f387614315a0 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 11 Mar 2025 22:29:06 -0700 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 25085dcd7a..1dd766c9b4 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -66,6 +66,7 @@ def unsloth_base_fast_generate( # VLMs do not allow logits_to_keep if not is_vlm: kwargs["logits_to_keep"] = 1 + print(kwargs) # Check pad_token model_eos_token_id = getattr(self.config, "eos_token_id", None)