Spaces:
Sleeping
Sleeping
Update musicgen_app.py
Browse files- musicgen_app.py +1 -1
musicgen_app.py
CHANGED
@@ -139,7 +139,7 @@ def _do_predictions(texts, melodies, duration, progress=False, **gen_kwargs):
|
|
139 |
return_tokens=USE_DIFFUSION
|
140 |
)
|
141 |
else:
|
142 |
-
outputs = MODEL.generate(
|
143 |
if USE_DIFFUSION:
|
144 |
outputs_diffusion = MBD.tokens_to_wav(outputs[1])
|
145 |
outputs = torch.cat([outputs[0], outputs_diffusion], dim=0)
|
|
|
139 |
return_tokens=USE_DIFFUSION
|
140 |
)
|
141 |
else:
|
142 |
+
outputs = MODEL.generate(newTexts, progress=progress, return_tokens=USE_DIFFUSION)
|
143 |
if USE_DIFFUSION:
|
144 |
outputs_diffusion = MBD.tokens_to_wav(outputs[1])
|
145 |
outputs = torch.cat([outputs[0], outputs_diffusion], dim=0)
|