optimize(_apply_spk_emb): expand after F.normalize (#495)

2noise · Jun 28, 2024 · 5c421c9 · 5c421c9
1 parent 5b537f3
commit 5c421c9
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ChatTTS/core.py b/ChatTTS/core.py
@@ -478,14 +478,14 @@ def _apply_spk_emb(
             F.normalize(
                 torch.from_numpy(
                     self._decode_spk_emb(spk_emb),
-                )
-                .unsqueeze(0)
-                .expand(text_len, -1),
+                ),
                 p=2.0,
-                dim=1,
+                dim=0,
                 eps=1e-12,
             )
             .to(self.gpt.device_gpt)
+            .unsqueeze_(0)
+            .expand(text_len, -1)
             .unsqueeze_(1)
             .expand(emb.shape)
         )