Spaces:

declare-lab
/

mustango

Runtime error

soujanyaporia commited on May 5, 2024

Commit

c963e98

verified ·

1 Parent(s): 19964e7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -226,11 +226,11 @@ class Mustango:
 # Initialize Mustango
 mustango = Mustango(device="cpu")
-mustango.vae.to(device_type)
-mustango.stft.to(device_type)
-mustango.model.to(device_type)
-mustango.music_model.beats_model.to(device_type)
-mustango.music_model.chords_model.to(device_type)
 # if torch.cuda.is_available():
 #     mustango = Mustango()
 # else:

 # Initialize Mustango
 mustango = Mustango(device="cpu")
+mustango.vae.to(device_selection)
+mustango.stft.to(device_selection)
+mustango.model.to(device_selection)
+mustango.music_model.beats_model.to(device_selection)
+mustango.music_model.chords_model.to(device_selection)
 # if torch.cuda.is_available():
 #     mustango = Mustango()
 # else: