Update vision.py

2025-05-15 04:26:05 -07:00 · 2025-05-15 04:26:05 -07:00 · a8884bc8c1
commit a8884bc8c1
parent e99dfc2589
1 changed files with 1 additions and 0 deletions
--- 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