futurespyhi commited on
Commit
bc268c7
·
1 Parent(s): 734b0b6

reduce repitition_penalty

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -255,6 +255,7 @@ def generate_music_spaces(lyrics: str, genre: str, mood: str, progress=gr.Progre
255
  "--max_new_tokens", "3000", # Full token count
256
  "--prompt_start_time", "0",
257
  "--prompt_end_time", "30", # Full 30-second clips
 
258
  ]
259
 
260
  # Use flash attention if available, otherwise fallback
 
255
  "--max_new_tokens", "3000", # Full token count
256
  "--prompt_start_time", "0",
257
  "--prompt_end_time", "30", # Full 30-second clips
258
+ "--repetition_penalty", "1.0"
259
  ]
260
 
261
  # Use flash attention if available, otherwise fallback