Revert "change g2p and other fix" (#40)

2023-09-30 09:13:39 +08:00
parent c3df0bab67
commit 2f687e4873
11 changed files with 253 additions and 75 deletions
--- a/models.py
+++ b/models.py
@@ -340,7 +340,7 @@ class TextEncoder(nn.Module):
        self.language_emb = nn.Embedding(num_languages, hidden_channels)
        nn.init.normal_(self.language_emb.weight, 0.0, hidden_channels**-0.5)
        self.bert_proj = nn.Conv1d(1024, hidden_channels, 1)
-        self.ja_bert_proj = nn.Conv1d(1024, hidden_channels, 1)
+        self.ja_bert_proj = nn.Conv1d(768, hidden_channels, 1)

        self.encoder = attentions.Encoder(
            hidden_channels,