fix: ensure SpkEncoder tensor with dtype.float

lenML · Jul 26, 2024 · b9491a6 · b9491a6
1 parent 3ce2424
commit b9491a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/core/spk/TTSSpeaker.py b/modules/core/spk/TTSSpeaker.py
@@ -50,7 +50,7 @@ class DcSpkEncoder(json.JSONEncoder):
     def default(self, obj):
         if isinstance(obj, torch.Tensor):
             # token 会触发这里
-            arr_bytes = obj.cpu().detach().numpy().tobytes()
+            arr_bytes = obj.cpu().float().detach().numpy().tobytes()
             return {
                 "_type": "torch.Tensor",
                 "data": base64.b64encode(arr_bytes).decode("utf-8"),