diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py
index e07f5ac2ce..629cee1c7e 100644
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@@ -59,6 +59,7 @@ __all__ = [
     "unpatch_unsloth_smart_gradient_checkpointing",
 
     "patch_compiled_autograd",
+    "process_vision_info",
 ]
 
 import torch
@@ -97,6 +98,9 @@ from unsloth_zoo.loss_utils import (
     HAS_CUT_CROSS_ENTROPY,
     fused_linear_cross_entropy,
 )
+from unsloth_zoo.vision_utils import (
+    process_vision_info,
+)
 
 # =============================================
 # Disable some warnings which can get annoying