diff --git a/README.md b/README.md index 281f5dd..e08b749 100644 --- 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"]) ```