fixed generation_args in Sample inference code
#73
by
dkleine
- opened
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": None,
|
| 137 |
"do_sample": False,
|
| 138 |
}
|
| 139 |
|