From abc50d7defe07ab4fc9b11883e90ae3236e7ed80 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 18 Mar 2025 01:33:38 -0700 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index d7467d820b..e0435d1ea0 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -118,6 +118,7 @@ def unsloth_base_fast_generate( key = NUM_LOGITS_TO_KEEP[arch] 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 if PROMPT_LOOPKUP[arch]: