diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 2bff87d8d9..83bd22f625 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -89,6 +89,7 @@ def unsloth_base_fast_generate( *args, **kwargs, ): + print(args, kwargs) if len(args) != 0: input_ids = args[0] elif "input_ids" in kwargs: