Update README.md

2024-07-26 13:07:36 +00:00
parent 0444474952
commit 62f26f7144
1 changed files with 1 additions and 1 deletions
--- a/README.md
+++ b/README.md
@@ -42,7 +42,7 @@ pipeline = transformers.pipeline(
    device_map="auto",
 )

-outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
+outputs = pipeline(prompt, max_new_tokens=2048, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
 print(outputs[0]["generated_text"])
 ```