Spaces:
Paused
Paused
Tao Wu commited on
Commit ·
a25d322
1
Parent(s): b2dd7b6
update exp prompt
Browse files- app/embedding_setup.py +1 -1
app/embedding_setup.py
CHANGED
|
@@ -171,7 +171,7 @@ def generate_exp(
|
|
| 171 |
top_p=1.0,
|
| 172 |
top_k=40,
|
| 173 |
num_beams=1,
|
| 174 |
-
max_new_tokens=
|
| 175 |
batch_size=1,
|
| 176 |
**kwargs,
|
| 177 |
):
|
|
|
|
| 171 |
top_p=1.0,
|
| 172 |
top_k=40,
|
| 173 |
num_beams=1,
|
| 174 |
+
max_new_tokens=512,
|
| 175 |
batch_size=1,
|
| 176 |
**kwargs,
|
| 177 |
):
|