Update README.md

README.md

@@ -62,13 +62,13 @@ tokenizer = transformers.AutoTokenizer.from_pretrained(model_name_or_path)
 input_text = "### Human: Give me a good recipe for a chinese dish\n\n### Assistant:"
 
 outputs = model.generate(
-    tokenizer(input_text, return_tensors="pt").to(model.device), max_length=1024,
+    tokenizer(input_text, return_tensors="pt").to(model.device),
+    max_length=1024,
     temperature=0.7,
     top_p=0.9,
     do_sample=True,
     pad_token_id=tokenizer.pad_token_id,
     eos_token_id=tokenizer.eos_token_id,
-    max_new_tokens=1024
 )
 print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 ```