rbyat commited on
Commit
543cf01
·
1 Parent(s): 11049d1

Update musicgen_app.py

Browse files
Files changed (1) hide show
  1. 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(newTextsexts, 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)
 
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)