Update README.md

README.md

@@ -82,7 +82,7 @@ def instruction_generator(system_message: str, num_instructions: int) -> str:
 	if num_instructions < 1:
 		raise ValueError
 	magpie_template = f"<|im_start|>system\n{system_message}<|im_end|>\n<|im_start|>user\n"
-	input_ids = tokenizer(danube_magpie, return_tensors="pt").input_ids.to("cuda")
+	input_ids = tokenizer(magpie_template, return_tensors="pt").input_ids.to("cuda")
 	for idx in range(num_instructions):
 		generated_ids = model.generate(input_ids, max_new_tokens=512, temperature=0.9, repetition_penalty=1.1, do_sample=True, eos_token_id=tokenizer.eos_token_id)
 		response = tokenizer.decode(generated_ids[0][input_ids.shape[-1]:], skip_special_tokens=True, clean_up_tokenization_space=True)