KingNish commited on
Commit
06ded55
·
verified ·
1 Parent(s): d21cf89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 * 100 tokens per second)
289
- max_new_tokens_scaled = max_new_tokens * 100
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: