From de7b9e87d795940df5dea4a5b5cdd9c35d7c0626 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 21 Nov 2024 03:05:13 -0800 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index f088797a8f..7d71bbf988 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -151,10 +151,6 @@ class FastBaseVisionModel: patch_saving_functions(model, vision = True) patch_saving_functions(tokenizer, vision = True) - # Fix gradient accumulation - from transformers.trainer import Trainer - patch_gradient_accumulation_fix(Trainer) - # Save tokenizer for inference purposes tokenizer.padding_side = "left" # Force inference internal_model = model