From 965880360fd44adfd8d1e5b477a71a40fd9ae1a6 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Tue, 18 Mar 2025 01:59:39 -0700
Subject: [PATCH] Update vision.py

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

diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py
index acf999faff..f4e220ca8a 100644
--- a/unsloth/models/vision.py
+++ b/unsloth/models/vision.py
@@ -92,6 +92,7 @@ def unsloth_base_fast_generate(
 
     FastBaseModel.for_inference(self)
     dtype = _get_dtype(self.config.torch_dtype)
+    print(dtype)
 
     # Check if VLM
     is_vlm = any(