Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ def generate(
|
|
47 |
max_new_tokens=max_new_tokens,
|
48 |
top_p=top_p,
|
49 |
repetition_penalty=repetition_penalty,
|
50 |
-
|
51 |
do_sample=True,
|
52 |
seed=seed,
|
53 |
)
|
|
|
47 |
max_new_tokens=max_new_tokens,
|
48 |
top_p=top_p,
|
49 |
repetition_penalty=repetition_penalty,
|
50 |
+
stop=STOP_SEQUENCES,
|
51 |
do_sample=True,
|
52 |
seed=seed,
|
53 |
)
|