Update vision.py

2025-05-15 03:54:22 -07:00 · 2025-05-15 03:54:22 -07:00 · dafdcbe88c
commit dafdcbe88c
parent ba6dcd2498
1 changed files with 1 additions and 0 deletions
--- a/unsloth/models/vision.py
+++ b/unsloth/models/vision.py
@ -212,6 +212,7 @@ def unsloth_base_fast_generate(
        PROMPT_LOOPKUP[arch] = False
        kwargs.pop("prompt_lookup_num_tokens", None)
        with torch.inference_mode(), autocaster:
+            print(args, kwargs)
            output = self._old_generate(*args, **kwargs)
    finally:
        pass