Spaces:
Runtime error
Runtime error
Commit
·
3b7f7a5
1
Parent(s):
1bbc4ff
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ def generate(prompt,history):
|
|
18 |
system_prompt = prompts.SONG_WRITER
|
19 |
generate_kwargs = dict(
|
20 |
temperature=0.9,
|
21 |
-
max_new_tokens=
|
22 |
top_p=0.95,
|
23 |
repetition_penalty=1.0,
|
24 |
do_sample=True,
|
|
|
18 |
system_prompt = prompts.SONG_WRITER
|
19 |
generate_kwargs = dict(
|
20 |
temperature=0.9,
|
21 |
+
max_new_tokens=4096,
|
22 |
top_p=0.95,
|
23 |
repetition_penalty=1.0,
|
24 |
do_sample=True,
|