diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py
index 86f25888e8..fd1f16e30b 100644
--- a/unsloth/models/llama.py
+++ b/unsloth/models/llama.py
@@ -1970,8 +1970,8 @@ class FastLlamaModel:
         # 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)
-            model.generate.__doc__ = model._old_generate.__doc__
         return model, tokenizer
     pass
 
@@ -2408,8 +2408,8 @@ class FastLlamaModel:
         # 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)
-            model.generate.__doc__ = model._old_generate.__doc__
         return model
     pass