diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py
index 525e562abb..39ff1fd2bb 100644
--- a/unsloth/models/vision.py
+++ b/unsloth/models/vision.py
@@ -366,7 +366,9 @@ class FastBaseModel:
                 # Execute code as well
                 if len(execute_code.strip()) != 0:
                     exec(execute_code)
-            pass
+            else:
+                custom_datatype = None
+                correct_dtype = None
         pass
 
         # Stop SDPA for some archs like Pixtral / Mistral3