Spaces:

sanchit-gandhi
/

musicgen-streaming

Running on Zero

sanchit-gandhi HF staff commited on Oct 13, 2023

Commit

9bd3413

•

1 Parent(s): e2c06c5

don't specify duration

Files changed (1) hide show

app.py CHANGED Viewed

@@ -176,7 +176,7 @@ target_dtype = np.int16
 max_range = np.iinfo(target_dtype).max
-@spaces.GPU(duration=30)
 def generate_audio(text_prompt, audio_length_in_s=10.0, play_steps_in_s=2.0, seed=0):
     max_new_tokens = int(frame_rate * audio_length_in_s)
     play_steps = int(frame_rate * play_steps_in_s)

 max_range = np.iinfo(target_dtype).max
+@spaces.GPU()
 def generate_audio(text_prompt, audio_length_in_s=10.0, play_steps_in_s=2.0, seed=0):
     max_new_tokens = int(frame_rate * audio_length_in_s)
     play_steps = int(frame_rate * play_steps_in_s)