Spaces:
Running
Running
futurespyhi
commited on
Commit
·
bc268c7
1
Parent(s):
734b0b6
reduce repitition_penalty
Browse files
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
|