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 |
|