Update README.md

README.md

@@ -28,8 +28,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from peft import PeftModel, PeftConfig
 
 # Load the base model
-model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7B-Instruct")
-tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7B-Instruct")
+model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
+tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
 
 # Load the LoRA adapter
 model = PeftModel.from_pretrained(model, "DDiaa/EXP-Llama-2-7B")