Spaces:
Running on Zero
Running on Zero
[Admin maintenance] Fix ZeroGPU
#106
by multimodalart HF Staff - opened
- demos/musicgen_app.py +1 -0
demos/musicgen_app.py
CHANGED
|
@@ -195,6 +195,7 @@ def _do_predictions(texts, melodies, duration, progress=False, gradio_progress=N
|
|
| 195 |
return out_wavs
|
| 196 |
|
| 197 |
|
|
|
|
| 198 |
def predict_batched(texts, melodies):
|
| 199 |
max_text_length = 512
|
| 200 |
texts = [text[:max_text_length] for text in texts]
|
|
|
|
| 195 |
return out_wavs
|
| 196 |
|
| 197 |
|
| 198 |
+
@spaces.GPU(duration=120)
|
| 199 |
def predict_batched(texts, melodies):
|
| 200 |
max_text_length = 512
|
| 201 |
texts = [text[:max_text_length] for text in texts]
|