From e24059fb037263f53c0634d1f7f178712bbac53b Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 18 Mar 2025 04:50:18 -0700 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 671734d28e..9e53176d52 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -146,7 +146,7 @@ def unsloth_base_fast_generate( try: kwargs["pixel_values"] = kwargs["pixel_values"].to(dtype) except: pass - print(args, kwargs) + print(args, kwargs, self._old_generate) # Mixed precision autocast if os.environ.get("UNSLOTH_FORCE_FLOAT32", "0") == "1":