update

app.py

@@ -17,7 +17,12 @@ from inversion_utils import inversion_forward_process, inversion_reverse_process
 LDM2 = "cvssp/audioldm2"
 MUSIC = "cvssp/audioldm2-music"
 LDM2_LARGE = "cvssp/audioldm2-large"
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+if torch.cuda.is_available():
+ device = "cuda"
+elif torch.backends.mps.is_available():
+ device = "mps"
+else:
+ device = "cpu"
 ldm2 = load_model(model_id=LDM2, device=device)
 ldm2_large = load_model(model_id=LDM2_LARGE, device=device)
 ldm2_music = load_model(model_id=MUSIC, device=device)

requirements.txt

@@ -1,5 +1,6 @@
-torch
-torchaudio
+gradio
+#torch
+#torchaudio
 diffusers
 accelerate
 transformers