From f0fc74cd9bf1f23ba725e3647026d0c86d7b71ef Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Tue, 18 Mar 2025 04:49:35 -0700
Subject: [PATCH] Update vision.py

---
 unsloth/models/vision.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py
index 0ab68a3e43..671734d28e 100644
--- a/unsloth/models/vision.py
+++ b/unsloth/models/vision.py
@@ -146,6 +146,8 @@ def unsloth_base_fast_generate(
     try: kwargs["pixel_values"] = kwargs["pixel_values"].to(dtype)
     except: pass
 
+    print(args, kwargs)
+
     # Mixed precision autocast
     if os.environ.get("UNSLOTH_FORCE_FLOAT32", "0") == "1":
         autocaster = torch.autocast(device_type = "cuda", dtype = dtype)