Spaces:
Runtime error
Runtime error
Fabrice-TIERCELIN
commited on
Commit
•
8a37844
1
Parent(s):
aff447f
OK, so uncomment
Browse files
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 |
-
|
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)
|