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