Update app.py
Browse files
app.py
CHANGED
@@ -285,8 +285,8 @@ def generate_music(
|
|
285 |
prompt_end_time=30.0,
|
286 |
rescale=False,
|
287 |
):
|
288 |
-
# Scale max_new_tokens (e.g. seconds *
|
289 |
-
max_new_tokens_scaled = max_new_tokens *
|
290 |
|
291 |
# Use a temporary directory to store intermediate stage outputs.
|
292 |
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
|
285 |
prompt_end_time=30.0,
|
286 |
rescale=False,
|
287 |
):
|
288 |
+
# Scale max_new_tokens (e.g. seconds * 50 tokens per second)
|
289 |
+
max_new_tokens_scaled = max_new_tokens * 50
|
290 |
|
291 |
# Use a temporary directory to store intermediate stage outputs.
|
292 |
with tempfile.TemporaryDirectory() as tmp_dir:
|