From f988ed485193f9de73ca0b9eac1ace43fc07f376 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Fri, 14 Mar 2025 05:41:17 -0700
Subject: [PATCH] Update vision.py

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

diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py
index 9c174e3a7d..e3fa946b7a 100644
--- a/unsloth/models/vision.py
+++ b/unsloth/models/vision.py
@@ -102,6 +102,8 @@ def unsloth_base_fast_generate(
         global NUM_LOGITS_TO_KEEP
         if arch not in NUM_LOGITS_TO_KEEP:
             m = self
+            # Find which is needed ie
+            # num_logits_to_keep or logits_to_keep
             while hasattr(m, "model"):
                 if hasattr(m, "forward"):
                     keys = inspect.signature(m.forward).parameters.keys()