Update app.py

app.py

@@ -169,6 +169,11 @@ class Mustango:
         stft_config = json.load(open(f"{path}/configs/stft_config.json"))
         main_config = json.load(open(f"{path}/configs/main_config.json"))
 
+        ALT_SD21 = "sd2-community/stable-diffusion-2-1"  # <-- mets TON repo alternatif exact
+
+        if main_config.get("scheduler_name") == "stabilityai/stable-diffusion-2-1":
+            main_config["scheduler_name"] = ALT_SD21
+
         self.vae = AutoencoderKL(**vae_config).to(device)
         self.stft = TacotronSTFT(**stft_config).to(device)
         self.model = MusicAudioDiffusion(