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