ZeroGPU

demos/musicgen_app.py

@@ -18,6 +18,7 @@ from tempfile import NamedTemporaryFile
 import time
 import typing as tp
 import warnings
+import spaces
 
 from einops import rearrange
 import torch
@@ -170,6 +171,7 @@ def _do_predictions(texts, melodies, duration, progress=False, gradio_progress=N
     return out_videos, out_wavs
 
 
+@spaces.GPU(duration=420)
 def predict_batched(texts, melodies):
     max_text_length = 512
     texts = [text[:max_text_length] for text in texts]
@@ -187,6 +189,7 @@ def check(model, model_path, decoder, text, melody, duration, topk, topp, temper
         raise gr.Error("Topp must not be negative.")
 
 
+@spaces.GPU(duration=420)
 def predict_full(model, model_path, decoder, text, melody, duration, topk, topp, temperature, cfg_coef, output_hint, progress=gr.Progress()):
     global INTERRUPTING
     global USE_DIFFUSION
@@ -267,8 +270,9 @@ def ui_full(launch_kwargs):
             The generated tracks tend to be very monotone so I advise you to add an original track to force the AI to make variations.
             <br/>
             <br/>
-            🐌 Slow process... ~6 hours for 2 minutes of music. Your computer must <b><u>not</u></b> enter into standby mode.
-            You can duplicate this space on a free account, it works on CPU.
+            """ + ("🏃‍♀️ Estimated time: few minutes." if torch.cuda.is_available() else "🐌 Slow process... ~6 hours for 2 minutes of music.") + """
+            Your computer must <b><u>not</u></b> enter into standby mode.
+            You can duplicate this space on a free account, it's designed to work on CPU, GPU and ZeroGPU.
             <a href="https://huggingface.co/spaces/Fabrice-TIERCELIN/Text-to-Music?duplicate=true"><img src="https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14"></a>
             <br/>
             ⚖️ If you use the standard models, you can use, modify and share the generated musics but not for commercial uses.