diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py
index efb1bcdb63..dbeb487e0d 100644
--- a/unsloth/models/vision.py
+++ b/unsloth/models/vision.py
@@ -113,7 +113,7 @@ def unsloth_base_fast_generate(
     except: pass
 
     # Mixed precision autocast
-    # if os.environ.get("UNSLOTH_FORCE_FLOAT32", "0") == "1": dtype = torch.float32
+    if os.environ.get("UNSLOTH_FORCE_FLOAT32", "0") == "1": dtype = torch.float32
     with torch.inference_mode(), torch.autocast(device_type = "cuda", dtype = dtype):
         output = self._old_generate(*args, **kwargs)
     pass