Update vision.py

2025-07-01 04:27:40 -07:00 · 2025-07-01 04:27:40 -07:00 · 551bb07bcc
commit 551bb07bcc
parent d35cf1633a
1 changed files with 1 additions and 1 deletions
--- a/unsloth/models/vision.py
+++ b/unsloth/models/vision.py
@ -562,7 +562,7 @@ class FastBaseModel:
        finetune_mlp_modules       = True,
        layers_to_transform        = None,
        layers_pattern             = None,
-        use_gradient_checkpointing = True,
+        use_gradient_checkpointing = "unsloth",
        random_state               = 3407,
        max_seq_length             = 2048, # not used anymore
        use_rslora                 = False,