Spaces:

declare-lab
/

mustango

Runtime error

soujanyaporia commited on May 5, 2024

Commit

1d02fbc

verified ·

1 Parent(s): b4eefd2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -229,6 +229,7 @@ mustango = Mustango(device="cpu")
 mustango.vae.to(device_type)
 mustango.stft.to(device_type)
 mustango.model.to(device_type)
 # if torch.cuda.is_available():
 #     mustango = Mustango()
 # else:

 mustango.vae.to(device_type)
 mustango.stft.to(device_type)
 mustango.model.to(device_type)
+mustango.music_model.to(device_type)
 # if torch.cuda.is_available():
 #     mustango = Mustango()
 # else: