Update README.md

README.md

@@ -41,10 +41,15 @@ with torch.no_grad():
     output_ids = model.generate(
         token_ids.to(model.device),
         max_new_tokens=256,
+        do_sample=True,
+        temperature=0.3,
+        top_p=0.95,
+        top_k=50,
+        repetition_penalty=1.1,
         pad_token_id=tokenizer.pad_token_id,
         eos_token_id=tokenizer.eos_token_id,
     )
-output = tokenizer.decode(output_ids.tolist()[0][token_ids.size(1) :], skip_special_tokens=True)
+output = tokenizer.decode(output_ids.tolist()[0][token_ids.size(1) :], skip_special_tokens=False)
 print(output)
 """