Update README.md

README.md

@@ -41,6 +41,6 @@ model = AutoModelForCausalLM.from_pretrained(checkpoint).half().to(device)
 instruction = "Write a function to scrape hacker news."
 prompt = f"Below is an instruction that describes a task.\n Write a response that appropriately completes the request.\n\n ### Instruction:\n{instruction}\n\n### Response:"
 inputs = tokenizer(prompt, return_tensors="pt").to(device)
-outputs = model.generate(**inputs,temperature=0.3,do_sample=True)
-print(tokenizer.decode(outputs[0],skip_special_tokens=True,max_new_tokens=200))
+outputs = model.generate(**inputs,temperature=0.3,do_sample=True,max_new_tokens=256)
+print(tokenizer.decode(outputs[0],skip_special_tokens=True))
 ```