Update README.md
Browse files
README.md
CHANGED
|
@@ -90,10 +90,10 @@ pipe = pipeline(
|
|
| 90 |
)
|
| 91 |
|
| 92 |
generation_args = {
|
| 93 |
-
"max_new_tokens":
|
| 94 |
"return_full_text": False,
|
| 95 |
-
"temperature": 0.
|
| 96 |
-
"do_sample":
|
| 97 |
}
|
| 98 |
|
| 99 |
output = pipe(messages, **generation_args)
|
|
|
|
| 90 |
)
|
| 91 |
|
| 92 |
generation_args = {
|
| 93 |
+
"max_new_tokens": 50,
|
| 94 |
"return_full_text": False,
|
| 95 |
+
"temperature": 0.2,
|
| 96 |
+
"do_sample": True,
|
| 97 |
}
|
| 98 |
|
| 99 |
output = pipe(messages, **generation_args)
|