Update app.py
Browse files
app.py
CHANGED
|
@@ -18,6 +18,8 @@ class Tango:
|
|
| 18 |
stft_config = json.load(open("{}/stft_config.json".format(path)))
|
| 19 |
main_config = json.load(open("{}/main_config.json".format(path)))
|
| 20 |
|
|
|
|
|
|
|
| 21 |
self.vae = AutoencoderKL(**vae_config).to(device)
|
| 22 |
self.stft = TacotronSTFT(**stft_config).to(device)
|
| 23 |
self.model = AudioDiffusion(**main_config).to(device)
|
|
|
|
| 18 |
stft_config = json.load(open("{}/stft_config.json".format(path)))
|
| 19 |
main_config = json.load(open("{}/main_config.json".format(path)))
|
| 20 |
|
| 21 |
+
main_config["scheduler_name"] = "sd2-community/stable-diffusion-2-1"
|
| 22 |
+
|
| 23 |
self.vae = AutoencoderKL(**vae_config).to(device)
|
| 24 |
self.stft = TacotronSTFT(**stft_config).to(device)
|
| 25 |
self.model = AudioDiffusion(**main_config).to(device)
|