Hugging Face's logo

Spaces:
declare-lab
/
mustango
Runtime error

App Files Community
40
soujanyaporia commited on
Commit
3ec605f
·
verified ·
1 Parent(s): 1d02fbc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -229,7 +229,8 @@ mustango = Mustango(device="cpu")
229
  mustango.vae.to(device_type)
230
  mustango.stft.to(device_type)
231
  mustango.model.to(device_type)
232
- mustango.music_model.to(device_type)
 
233
  # if torch.cuda.is_available():
234
  # mustango = Mustango()
235
  # else:
 
229
  mustango.vae.to(device_type)
230
  mustango.stft.to(device_type)
231
  mustango.model.to(device_type)
232
+ mustango.music_model.beats_model.to(device_type)
233
+ mustango.music_model.chord_model.to(device_type)
234
  # if torch.cuda.is_available():
235
  # mustango = Mustango()
236
  # else: