Export kokoro 1.0 int8 models (#2137)

2025-04-20 14:35:02 +08:00
parent be0f382a54
commit 6cabaa11bf
7 changed files with 212 additions and 99 deletions
--- a/scripts/kokoro/v1.0/add_meta_data.py
+++ b/scripts/kokoro/v1.0/add_meta_data.py
@@ -10,7 +10,9 @@ from generate_voices_bin import speaker2id

 def main():
    model = onnx.load("./kokoro.onnx")
-    style = torch.load("./voices/af_alloy.pt", weights_only=True, map_location="cpu")
+    style = torch.load(
+        "./Kokoro-82M/voices/af_alloy.pt", weights_only=True, map_location="cpu"
+    )

    id2speaker_str = ""
    speaker2id_str = ""