Fabrice-TIERCELIN commited on
Commit
8a37844
1 Parent(s): aff447f

OK, so uncomment

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -48,8 +48,8 @@ class Tango:
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)
54
  # stft_weights = torch.load("{}/pytorch_model_stft.bin".format(path), map_location = device)
55
  # main_weights = torch.load("{}/pytorch_model_main.bin".format(path), map_location = device)
 
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)
54
  # stft_weights = torch.load("{}/pytorch_model_stft.bin".format(path), map_location = device)
55
  # main_weights = torch.load("{}/pytorch_model_main.bin".format(path), map_location = device)