Update llama.py did not included rotary embeddings in the reset funct…

…ions correctly

unsloth/models/llama.py

@@ -1511,6 +1511,10 @@ def _fast_generate(*args, **kwargs):
 original_llama_model_forward          = LlamaModel.forward
 original_llama_for_causal_lm_forward  = LlamaForCausalLM.forward
 original_peft_model_for_causal_lm_forward = PeftModelForCausalLM.forward
+import transformers.models.llama.modeling_llama
+original_LLamaRotaryEmbedding =  transformers.models.llama.modeling_llama.LlamaRotaryEmbedding 
+original_LLamaLinearScalingRotaryEmbedding = transformers.models.llama.modeling_llama.LlamaLinearScalingRotaryEmbedding
+
 class FastLlamaModel:
     def set_functions():
         LlamaAttention      .forward = LlamaAttention_fast_forward