Spaces:
Running on Zero
Running on Zero
Commit ·
39609d0
1
Parent(s): 08029ce
[Admin maintenance] Fix ZeroGPU (#106)
Browse files- [Admin maintenance] Fix ZeroGPU (999f6706dc5e878a3f14e7305e236e2ec5fb401e)
- 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]
|