fix: _encode_spk_emb use 'tobytes' without check 'dtype' (#480)

2noise · Jun 27, 2024 · 753f5f1 · 753f5f1
1 parent 48bf2ac
commit 753f5f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ChatTTS/core.py b/ChatTTS/core.py
@@ -162,7 +162,7 @@ def sample_random_speaker(self) -> str:
     @staticmethod
     def _encode_spk_emb(spk_emb: torch.Tensor) -> str:
         with torch.no_grad():
-            arr: np.ndarray = spk_emb.cpu().numpy()
+            arr: np.ndarray = spk_emb.to(dtype=torch.float16, device='cpu').numpy()
             s = b14.encode_to_string(
                 lzma.compress(
                     arr.tobytes(),