新增gpu模式下chattts代码

Chat2TTS/core.py

@@ -175,6 +175,21 @@ class Chat:
         return self._encode_spk_emb(self._sample_random_speaker())
 
 
+    @staticmethod
+    def _encode_spk_emb(spk_emb: torch.Tensor) -> str:
+        with torch.no_grad():
+            arr: np.ndarray = spk_emb.to(dtype=torch.float16, device="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
+        return s
 
     def _sample_random_speaker(self) -> torch.Tensor:
         with torch.no_grad():