Spaces:

thepatch
/

stable-melodyflow

Sleeping

thecollabagepatch commited on Jul 12

Commit

c3af266

1 Parent(s): 3e6fdea

trying to solve model loading

Files changed (1) hide show

app.py CHANGED Viewed

@@ -89,7 +89,7 @@ def load_stable_audio_model():
                 model_cache['stable_audio_config'],
                 model_cache['stable_audio_device'])
-@spaces.GPU
 def generate_stable_audio_loop(prompt, loop_type, bpm, bars, seed=-1):
     """Generate a BPM-aware loop using stable-audio-open-small"""
     try:

                 model_cache['stable_audio_config'],
                 model_cache['stable_audio_device'])
+@spaces.GPU(duration=12)
 def generate_stable_audio_loop(prompt, loop_type, bpm, bars, seed=-1):
     """Generate a BPM-aware loop using stable-audio-open-small"""
     try: