Spaces:
Runtime error
Runtime error
Fabrice-TIERCELIN
commited on
Commit
•
aff447f
1
Parent(s):
6dbb98e
OK -> Continue to uncomment
Browse files
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 |
-
|
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)
|