test

app.py

@@ -25,13 +25,11 @@ from peft import PeftModel
 from peft import get_peft_model
 from peft import set_peft_model_state_dict
 import transformers
-from transformers import LlamaForCausalLM
-from transformers import LlamaTokenizer
 from transformers import PreTrainedModel
 from transformers import PreTrainedTokenizerBase
 
-transformers.AutoTokenizer.from_pretrained = LlamaTokenizer.from_pretrained
-transformers.AutoModelForCausalLM.from_pretrained = LlamaForCausalLM.from_pretrained
+transformers.AutoTokenizer = transformers.LlamaTokenizer
+transformers.AutoModelForCausalLM = transformers.LlamaForCausalLM
 
 def load_lora_model(
     model_path: str,