feat(chat): add hidden api _encode_spk_emb (#474)

2noise · Jun 27, 2024 · bb9d3b3 · bb9d3b3
1 parent f8f4f53
commit bb9d3b3
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/ChatTTS/core.py b/ChatTTS/core.py
@@ -154,17 +154,20 @@ def unload(self):
         self.__init__(logger)
 
     def sample_random_speaker(self) -> str:
+        return self._encode_spk_emb(self._sample_random_speaker())
+
+    @staticmethod
+    def _encode_spk_emb(spk_emb: torch.Tensor) -> str:
         with torch.no_grad():
-            spk = self._sample_random_speaker()
-            arr: np.ndarray = spk.cpu().numpy()
+            arr: np.ndarray = spk_emb.cpu().numpy()
             s = b14.encode_to_string(
                 lzma.compress(
                     arr.tobytes(),
                     format=lzma.FORMAT_RAW,
                     filters=[{"id": lzma.FILTER_LZMA2, "preset": 9 | lzma.PRESET_EXTREME}],
                 ),
             )
-            del arr, spk
+            del arr
         return s
 
     def _sample_random_speaker(self) -> torch.Tensor: