first

2026-08-26 17:42:00 -04:00
commit ad95d530ad
134 changed files with 31750 additions and 0 deletions
@@ -0,0 +1,38 @@
+from dataclasses import dataclass, field
+from typing import Optional
+
+
+@dataclass
+class KokoroTTSHandlerArguments:
+    kokoro_model_name: Optional[str] = field(
+        default=None,
+        metadata={
+            "help": "The Kokoro TTS model to use. Auto-selects based on device: 'mlx-community/Kokoro-82M-bf16' for MPS, 'hexgrad/Kokoro-82M' for CUDA/CPU."
+        },
+    )
+    kokoro_device: str = field(
+        default="auto",
+        metadata={
+            "help": "The device to run Kokoro TTS on. Options: 'auto', 'cuda', 'cpu', 'mps'. Default is 'auto' (MPS on Mac, CUDA on GPU systems)."
+        },
+    )
+    kokoro_voice: str = field(
+        default="bm_fable",
+        metadata={
+            "help": "The voice to use for synthesis. See VOICES.md in the Kokoro repo for options. Default is 'bm_fable' (British male)."
+        },
+    )
+    kokoro_lang_code: str = field(
+        default="b",
+        metadata={
+            "help": "Language code: 'a' for American English, 'b' for British English, 'j' for Japanese, etc. Default is 'b'."
+        },
+    )
+    kokoro_speed: float = field(
+        default=1.0,
+        metadata={"help": "Speech speed multiplier. Values > 1.0 speed up, < 1.0 slow down. Default is 1.0."},
+    )
+    kokoro_blocksize: int = field(
+        default=512,
+        metadata={"help": "The audio chunk size in samples for streaming output. Default is 512."},
+    )