fffiloni commited on
Commit
0945840
·
verified ·
1 Parent(s): 8cea530

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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)