Merge branch 'dev-0309-training' of github.com-zwl:myshell-ai/MeloTTS into dev-0309-training

This commit is contained in:
wl-zhao
2024-03-16 16:02:53 +00:00

View File

@@ -25,9 +25,9 @@ DOWNLOAD_CONFIG_URLS = {
}
PRETRAINED_MODELS = {
'G.pth': 'https://cloud.tsinghua.edu.cn/f/91346812c11746e1b67b/?dl=1',
'D.pth': 'https://cloud.tsinghua.edu.cn/f/4734a5281219424199f1/?dl=1',
'DUR.pth': 'https://cloud.tsinghua.edu.cn/f/750feac7585f49ce96d7/?dl=1',
'G.pth': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/pretrained/G.pth',
'D.pth': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/pretrained/D.pth',
'DUR.pth': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/pretrained/DUR.pth',
}
LANG_TO_HF_REPO_ID = {
@@ -63,4 +63,4 @@ def load_or_download_model(locale, device, use_hf=True, ckpt_path=None):
return torch.load(ckpt_path, map_location=device)
def load_pretrain_model():
return [cached_path(url) for url in PRETRAINED_MODELS.values()]
return [cached_path(url) for url in PRETRAINED_MODELS.values()]