Spaces:

sanchit-gandhi
/

musicgen-streaming

Running on Zero

sanchit-gandhi HF staff commited on Apr 15

Commit

fa5be88

•

1 Parent(s): 6f05f79

Update app.py

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()
 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)