Skip to content

Commit

Permalink
fix: update kokoro-onnx to version 1.0
Browse files Browse the repository at this point in the history
Co-Authored-By: Nicolas Arqueros <nico@dcspark.io>
  • Loading branch information
devin-ai-integration[bot] and nicarq committed Feb 8, 2025
1 parent a733752 commit 74ddbfe
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tools/text-to-audio-kokoro/tool.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,8 +24,8 @@
config.MAX_PHONEME_LENGTH = 128

# URLs from the official kokoro-onnx repository
MODEL_URL = "https://github.com/thewh1teagle/kokoro-onnx/releases/download/model-files/kokoro-v0_19.onnx"
VOICES_URL = "https://github.com/thewh1teagle/kokoro-onnx/releases/download/model-files/voices.bin"
MODEL_URL = "https://github.com/thewh1teagle/kokoro-onnx/releases/download/v1.0/kokoro-v1_0.onnx"
VOICES_URL = "https://github.com/thewh1teagle/kokoro-onnx/releases/download/v1.0/voices.bin"

def download_file(url: str, local_path: str) -> None:
print(f"Downloading {url}...")
Expand All @@ -49,7 +49,7 @@ def download_file(url: str, local_path: str) -> None:
print("\nDownload complete!")

class CONFIG:
model_path: str = "kokoro-v0_19.onnx"
model_path: str = "kokoro-v1_0.onnx"
voices_path: str = "voices.bin"
providers: Optional[List[str]] = None # ONNX providers (e.g. ["CPUExecutionProvider", "CUDAExecutionProvider"])

Expand Down Expand Up @@ -136,4 +136,4 @@ async def run(c: CONFIG, p: INPUTS) -> OUTPUT:
output.sample_rate = sample_rate
output.chars_per_second = chars_per_second

return output
return output

0 comments on commit 74ddbfe

Please sign in to comment.