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