Update README.md
Browse files
README.md
CHANGED
@@ -97,7 +97,7 @@ prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_
|
|
97 |
# print("***Prompt:\n", prompt)
|
98 |
|
99 |
outputs = pipeline(
|
100 |
-
prompt, max_new_tokens=1000, do_sample=True, temperature=0.
|
101 |
)
|
102 |
print("***Generation:")
|
103 |
print(outputs[0]["generated_text"][len(prompt) :])
|
|
|
97 |
# print("***Prompt:\n", prompt)
|
98 |
|
99 |
outputs = pipeline(
|
100 |
+
prompt, max_new_tokens=1000, do_sample=True, temperature=0.7, top_k=50, top_p=0.95
|
101 |
)
|
102 |
print("***Generation:")
|
103 |
print(outputs[0]["generated_text"][len(prompt) :])
|