From aca62cfa0122dcb462ae773631937c8f6da76c9d Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Tue, 18 Mar 2025 03:45:55 -0700
Subject: [PATCH] Update vision.py

---
 unsloth/models/vision.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py
index acf999faff..0ab68a3e43 100644
--- a/unsloth/models/vision.py
+++ b/unsloth/models/vision.py
@@ -126,8 +126,8 @@ def unsloth_base_fast_generate(
         pass
     pass
     key = NUM_LOGITS_TO_KEEP[arch]
-    if key is not None and key not in kwargs:
-        kwargs[key] = 1
+    # if key is not None and key not in kwargs:
+    #     kwargs[key] = 1
     global PROMPT_LOOPKUP
     if arch not in PROMPT_LOOPKUP:
         PROMPT_LOOPKUP[arch] = True