broadfield commited on
Commit
3b7f7a5
·
1 Parent(s): 1bbc4ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=1024,
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,