Fabrice-TIERCELIN commited on
Commit
f9b54be
1 Parent(s): 05f5a49

Still KO -> more comment

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