From db1c5f414a2c23d2ec0336927adba69c30e0bdae Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Sun, 17 Nov 2024 16:56:26 -0800
Subject: [PATCH] Update _utils.py

---
 unsloth/models/_utils.py | 4 ++++
 1 file changed, 4 insertions(+)

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