From d42f11f41e8c014a12dbe570eea3e7f8bdc031be Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Fri, 14 Mar 2025 06:11:51 -0700
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 616cc8fe59..cd953363c4 100644
--- a/unsloth/models/vision.py
+++ b/unsloth/models/vision.py
@@ -125,7 +125,7 @@ def unsloth_base_fast_generate(
     else:
         kwargs.pop("logits_to_keep", None)
         kwargs.pop("num_logits_to_keep", None)
-    kwargs["logits_to_keep"] = 0
+    kwargs["num_logits_to_keep"] = 0
 
     # Check pad_token
     model_eos_token_id = getattr(self.config, "eos_token_id", None)