diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index 5aabf15134..88494df3d2 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -2665,13 +2665,6 @@ class FastLlamaModel: model.load_lora = functools.partial(load_lora, model) pass - # Patch generate - if model.generate.__name__ != "unsloth_fast_generate": - model._old_generate = model.generate - unsloth_fast_generate.__doc__ = model._old_generate.__doc__ - model.generate = types.MethodType(unsloth_fast_generate, model) - pass - # Add for_inference and for_training model.for_training = functools.partial(FastLlamaModel.for_training, model) model.for_inference = functools.partial(FastLlamaModel.for_inference, model)