修复了VQ未注册入网络的问题 (#216)

* Add files via upload

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

models.py

@@ -345,16 +345,18 @@ class TextEncoder(nn.Module):
         self.ja_bert_proj = nn.Conv1d(1024, hidden_channels, 1)
         self.en_bert_proj = nn.Conv1d(1024, hidden_channels, 1)
         self.emo_proj = nn.Linear(1024, 1024)
-        self.emo_quantizer = [
+        self.emo_quantizer = nn.ModuleList()
-            VectorQuantize(
+        for i in range(0, n_speakers):
-                dim=1024,
+            self.emo_quantizer.append(
-                codebook_size=10,
+                VectorQuantize(
-                decay=0.8,
+                    dim=1024,
-                commitment_weight=1.0,
+                    codebook_size=10,
-                learnable_codebook=True,
+                    decay=0.8,
-                ema_update=False,
+                    commitment_weight=1.0,
+                    learnable_codebook=True,
+                    ema_update=False,
+                )
             )
-        ] * n_speakers
         self.emo_q_proj = nn.Linear(1024, hidden_channels)
 
         self.encoder = attentions.Encoder(