From ea20b0a2fe8a9bc936cbf9ff54a1a937691eaed2 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 18 Mar 2025 22:46:30 -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 acf999faff..f30117c917 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -543,10 +543,10 @@ class FastBaseModel: model.for_inference = functools.partial(FastBaseModel.for_inference, model) # Patch generate - if model.generate.__name__ != "unsloth_base_fast_generate": - model._old_generate = model.generate - unsloth_base_fast_generate.__doc__ = model._old_generate.__doc__ - model.generate = types.MethodType(unsloth_base_fast_generate, model) + # if model.generate.__name__ != "unsloth_base_fast_generate": + # model._old_generate = model.generate + # unsloth_base_fast_generate.__doc__ = model._old_generate.__doc__ + # model.generate = types.MethodType(unsloth_base_fast_generate, model) return model pass