fixed generation_args in Sample inference code
Browse files* fixed temperature for "do_sample": False (`temperature=0.0` would require sampling, which is turned off with `"do_sample": False`. therefore better to use `temperature=1.0` without sampling)
README.md
CHANGED
|
@@ -133,7 +133,7 @@ pipe = pipeline(
|
|
| 133 |
generation_args = {
|
| 134 |
"max_new_tokens": 500,
|
| 135 |
"return_full_text": False,
|
| 136 |
-
"temperature":
|
| 137 |
"do_sample": False,
|
| 138 |
}
|
| 139 |
|
|
|
|
| 133 |
generation_args = {
|
| 134 |
"max_new_tokens": 500,
|
| 135 |
"return_full_text": False,
|
| 136 |
+
"temperature": 1.0,
|
| 137 |
"do_sample": False,
|
| 138 |
}
|
| 139 |
|