diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py
index a65b538745..4df922fce0 100644
--- a/unsloth/models/vision.py
+++ b/unsloth/models/vision.py
@@ -42,6 +42,7 @@ from unsloth_zoo.patching_utils import patch_model_and_tokenizer
 from unsloth_zoo.training_utils import prepare_model_for_training
 import types
 import functools
+import os
 
 __all__ = [
     "FastBaseModel",