Update vision.py

2025-03-13 01:27:50 -07:00 · 2025-03-13 01:27:50 -07:00 · ba096eda6e
commit ba096eda6e
parent 7bd582a6ff
1 changed files with 4 additions and 4 deletions
--- a/unsloth/models/vision.py
+++ b/unsloth/models/vision.py
@ -297,10 +297,10 @@ class FastBaseModel:
            model.generate = types.MethodType(unsloth_base_fast_generate, model)

        # Post patches
-        model = FastBaseModel.post_patch_model(
-            model,
-            use_gradient_checkpointing = use_gradient_checkpointing,
-        )
+        # model = FastBaseModel.post_patch_model(
+        #     model,
+        #     use_gradient_checkpointing = use_gradient_checkpointing,
+        # )
        # Clear deleted GPU items
        for _ in range(3):
            gc.collect()