From 4ca856fefdbff30b6f8812f625602f46485a5fd6 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Fri, 14 Mar 2025 05:54:57 -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 e3fa946b7a..7993a9a48e 100644
--- a/unsloth/models/vision.py
+++ b/unsloth/models/vision.py
@@ -137,7 +137,7 @@ def unsloth_base_fast_generate(
     try: kwargs["pixel_values"] = kwargs["pixel_values"].to(dtype)
     except: pass
 
-    print(kwargs)
+    print(kwargs.keys())
 
     # Mixed precision autocast
     if os.environ.get("UNSLOTH_FORCE_FLOAT32", "0") == "1": dtype = torch.float32