From ba096eda6e228af7993cd6154834d497174da8e5 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 13 Mar 2025 01:27:50 -0700 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 73497e70a7..fc9519cb64 100644 --- 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()