Fabrice-TIERCELIN commited on
Commit
aff447f
1 Parent(s): 6dbb98e

OK -> Continue to uncomment

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ class Tango:
47
  main_config = json.load(open("{}/main_config.json".format(path)))
48
 
49
  self.vae = AutoencoderKL(**vae_config).to(device)
50
- # self.stft = TacotronSTFT(**stft_config).to(device)
51
  # self.model = AudioDiffusion(**main_config).to(device)
52
  #
53
  # vae_weights = torch.load("{}/pytorch_model_vae.bin".format(path), map_location = device)
 
47
  main_config = json.load(open("{}/main_config.json".format(path)))
48
 
49
  self.vae = AutoencoderKL(**vae_config).to(device)
50
+ self.stft = TacotronSTFT(**stft_config).to(device)
51
  # self.model = AudioDiffusion(**main_config).to(device)
52
  #
53
  # vae_weights = torch.load("{}/pytorch_model_vae.bin".format(path), map_location = device)