Update README.md

README.md

@@ -38,7 +38,7 @@ model     = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=compute_d
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 chat    = tokenizer.apply_chat_template([{"role":"user", "content":"What is 1.5+102.2?"}], tokenize=True, add_generation_prompt=True, return_tensors="pt")
-outputs = model_student.generate(chat.to(device), max_new_tokens=1024, do_sample=True) 
+outputs = model.generate(chat.to(device), max_new_tokens=1024, do_sample=True) 
 print(tokenizer.decode(outputs[0]))
 ```