From f89f30aa24b8396f47909e2606db9e28ca347150 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Fri, 14 Mar 2025 06:17:35 -0700 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index ac4c6287e8..31733c2976 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -120,9 +120,8 @@ def unsloth_base_fast_generate( pass pass key = NUM_LOGITS_TO_KEEP[arch] - if key is not None: - if key not in kwargs: - kwargs[key] = 1 + if key is not None and key not in kwargs: + kwargs[key] = 1 else: pass # kwargs.pop("logits_to_keep", None)