Update README.md

This commit is contained in:
Taeyoung Lee
2024-07-26 13:07:36 +00:00
committed by system
parent 0444474952
commit 62f26f7144

View File

@@ -42,7 +42,7 @@ pipeline = transformers.pipeline(
device_map="auto", 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"]) print(outputs[0]["generated_text"])
``` ```