Update README.md

README.md

@@ -21,7 +21,7 @@ Use the code below to get started with the model.
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from peft import PeftModel
 
-base_model = AutoModelForCausalLM.from_pretrained('meta-llama/Llama-2-7b')
+base_model = AutoModelForCausalLM.from_pretrained('meta-llama/Llama-2-7b-hf')
 base_model.resize_token_embeddings(260164)
 tokenizer = AutoTokenizer.from_pretrained('MaLA-LM/mala-500')
 model = PeftModel.from_pretrained(base_model, 'MaLA-LM/mala-500')