futurespyhi commited on
Commit
1f93c91
·
1 Parent(s): 7e55e4f

add GPU running time for less powerful GPUs

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -287,7 +287,7 @@ def validate_file_structure():
287
  print("✅ All required files found")
288
  return True
289
 
290
- @spaces.GPU(duration=900) # Increase GPU timeout to 15 minutes
291
  def generate_music_spaces(lyrics: str, genre: str, mood: str, progress=gr.Progress()) -> str:
292
  """
293
  Generate music using YuE model with high-performance Spaces configuration
 
287
  print("✅ All required files found")
288
  return True
289
 
290
+ @spaces.GPU(duration=1200) # Increase GPU timeout to 20 minutes for other less powerful GPUs
291
  def generate_music_spaces(lyrics: str, genre: str, mood: str, progress=gr.Progress()) -> str:
292
  """
293
  Generate music using YuE model with high-performance Spaces configuration