deepgeek/Matcha-TTS
2
0
Fork 0
mirror of https://github.com/shivammehta25/Matcha-TTS.git synced 2026-02-05 18:29:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

adding more validation to multispeaker CLI

Browse Source
This commit is contained in:
Shivam Mehta
2023-09-18 11:37:58 +00:00
parent ec43ef0732
commit d7b9a37359
2 changed files with 32 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
matcha/app.py
View File

@@ -8,7 +8,7 @@ import torch
from matcha.cli import (
MATCHA_URLS,
VOCODER_URL,
VOCODER_URLS,
assert_model_downloaded,
get_device,
load_matcha,
@@ -31,7 +31,7 @@ MATCHA_TTS_LOC = LOCATION / f"{args.model}.ckpt"
VOCODER_LOC = LOCATION / f"{args.vocoder}"
LOGO_URL = "https://shivammehta25.github.io/Matcha-TTS/images/logo.png"
assert_model_downloaded(MATCHA_TTS_LOC, MATCHA_URLS[args.model])
assert_model_downloaded(VOCODER_LOC, VOCODER_URL[args.vocoder])
assert_model_downloaded(VOCODER_LOC, VOCODER_URLS[args.vocoder])
device = get_device(args)
model = load_matcha(args.model, MATCHA_TTS_LOC, device)