Update vision.py

2025-09-18 22:33:13 -07:00 · 2025-09-18 22:33:13 -07:00 · af60134d7d
commit af60134d7d
parent 5d6fbda29a
1 changed files with 1 additions and 0 deletions
--- a/unsloth/models/vision.py
+++ b/unsloth/models/vision.py
@ -805,6 +805,7 @@ class FastBaseModel:
            train_embedding            = full_finetuning,
            train_lm_head              = full_finetuning,
            float32_mixed_precision    = float32_mixed_precision,
+            patch_modules_to_save      = True,
        )

        from transformers.trainer import Trainer