multimodalart HF Staff commited on
Commit
999f670
·
verified ·
1 Parent(s): 08029ce

[Admin maintenance] Fix ZeroGPU

Browse files

Thank you so much for having shared this Space with the community on this demo. We have upgraded the ZeroGPU infra-structure to run on modern blackwell architecture.
For that, we need to upgrade your demo to support that. This PR fixes your demo to work with the new architecture. As this is something we broke on our end, we may merge this PR autonomously. If this breaks unexpectedly or brings unintended consequences, feel free to revert, modify or otherwise. Any issues you can email apolinario@huggingface.co

Files changed (1) hide show
  1. 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]