diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py
index 9f847f2837..b4ce718f46 100644
--- a/unsloth/models/vision.py
+++ b/unsloth/models/vision.py
@@ -930,7 +930,7 @@ class FastBaseModel:
         task_type = TaskType.CAUSAL_LM,
         temporary_location = "_unsloth_temporary_saved_buffers",
         qat_scheme = None,
-        ensure_weight_tying = False,
+        ensure_weight_tying = False,  # [TODO] Add `ensure_weight_tying` for `modules_to_save` for vision models
         **kwargs,
     ):
         if os.environ.get("UNSLOTH_ENABLE_FULL_FINETUNING", "0") == "1":