coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Save speakers embeddings/ids before starting training

pull/1032/head
Edresson 2021-08-12 07:52:44 -03:00 committed by Eren Gölge
parent 37803467aa
commit eb3e8affe1
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
TTS/tts/models/vits.py
View File

@ -402,7 +402,6 @@ class Vits(BaseTTS):
# speaker embedding # speaker embedding
if self.num_speakers > 1 and sid is not None: if self.num_speakers > 1 and sid is not None:
g = self.emb_g(sid).unsqueeze(-1) # [b, h, 1] g = self.emb_g(sid).unsqueeze(-1) # [b, h, 1]
# posterior encoder # posterior encoder
z, m_q, logs_q, y_mask = self.posterior_encoder(y, y_lengths, g=g) z, m_q, logs_q, y_mask = self.posterior_encoder(y, y_lengths, g=g)

2
requirements.txt
View File

@ -13,7 +13,7 @@ pyyaml
scipy>=0.19.0 scipy>=0.19.0
soundfile soundfile
tensorboardX tensorboardX
torch>=1.7 torch>=1.9.0
tqdm tqdm
numba==0.53 numba==0.53
umap-learn==0.5.1 umap-learn==0.5.1