Spaces:
Build error
Build error
updates to return tokens
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ def text_generate(problem, template, prompt):
|
|
| 37 |
{
|
| 38 |
"top_p": 0.9,
|
| 39 |
"temperature": 1.1,
|
| 40 |
-
"max_new_tokens":
|
| 41 |
"return_full_text": True
|
| 42 |
}, "options":
|
| 43 |
{
|
|
|
|
| 37 |
{
|
| 38 |
"top_p": 0.9,
|
| 39 |
"temperature": 1.1,
|
| 40 |
+
"max_new_tokens": 64,
|
| 41 |
"return_full_text": True
|
| 42 |
}, "options":
|
| 43 |
{
|