Spaces:

declare-lab
/

mustango

Running on A10G

soujanyaporia commited on May 6

Commit

cfc99ed

•

1 Parent(s): b59693e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -225,19 +225,19 @@ 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(device=device_selection)
 # else:
 #     mustango = Mustango(device="CPU")
-mustango = Mustango(device=device_selection)
 mustango.model.device = device_selection

 # 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(device=device_selection)
 # else:
 #     mustango = Mustango(device="CPU")
+# mustango = Mustango(device=device_selection)
 mustango.model.device = device_selection