From a8884bc8c12deb8036249cb340cad824030f5e4e Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 15 May 2025 04:26:05 -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 782c30fcf9..93c72ec312 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -207,6 +207,7 @@ def unsloth_base_fast_generate( try: with torch.inference_mode(), autocaster: + print(args, kwargs) output = self._old_generate(*args, **kwargs) except: PROMPT_LOOPKUP[arch] = False