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