Update README.md
Browse files
README.md
CHANGED
@@ -33,6 +33,6 @@ pipeline = transformers.pipeline(
|
|
33 |
|
34 |
messages = [{"role": "user", "content": "Tell me about AI"}]
|
35 |
prompt = pipeline.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
36 |
-
outputs = pipeline(prompt, max_new_tokens=
|
37 |
print(outputs[0]["generated_text"])
|
38 |
```
|
|
|
33 |
|
34 |
messages = [{"role": "user", "content": "Tell me about AI"}]
|
35 |
prompt = pipeline.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
36 |
+
outputs = pipeline(prompt, max_new_tokens=1024, do_sample=True, temperature=0.7, top_k=500, top_p=0.95)
|
37 |
print(outputs[0]["generated_text"])
|
38 |
```
|