Update README.md

README.md

@@ -53,12 +53,12 @@ from transformers import AutoTokenizer, AutoModelModelForCausalLM
 
 # load base LLM model and tokenizer
 model = AutoModelModelForCausalLM.from_pretrained(
-    "philschmid/llama-7b-instruction-generator",
+    "philschmid/llama-2-7b-instruction-generator",
     low_cpu_mem_usage=True,
     torch_dtype=torch.float16,
     load_in_4bit=True,
 ) 
-tokenizer = AutoTokenizer.from_pretrained("philschmid/llama-7b-instruction-generator")
+tokenizer = AutoTokenizer.from_pretrained("philschmid/llama-2-7b-instruction-generator")
 
 prompt = f"""### Instruction:
 Use the Input below to create an instruction, which could have been used to generate the input using an LLM.