Propose Zero GPU

demos/musicgen_app.py

@@ -22,6 +22,7 @@ import warnings
 from einops import rearrange
 import torch
 import gradio as gr
+import spaces
 
 from audiocraft.data.audio_utils import convert_audio
 from audiocraft.data.audio import audio_write
@@ -170,6 +171,7 @@ def _do_predictions(texts, melodies, duration, progress=False, gradio_progress=N
     return out_videos, out_wavs
 
 
+@spaces.GPU
 def predict_batched(texts, melodies):
     max_text_length = 512
     texts = [text[:max_text_length] for text in texts]