Update README.md

README.md

@@ -48,7 +48,7 @@ article = ""  # insert a document here
 prompt = f"### Human: Please summarize the following article.\n\n{article}.\n###"
 
 inputs = tokenizer(header + prompt, return_tensors="pt")
-sample = model.generate(**inputs, max_new_tokens=128)
+sample = model.generate(**inputs, do_sample=True, max_new_tokens=2048, top_k=100, eos_token_id=50256)
 print(tokenizer.decode(sample[0]))
 ```