zhangshuai/MeloTTS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use cached_path for better caching

Browse Source
This commit is contained in:
mrfakename
2024-02-26 15:53:10 -08:00
committed by GitHub
parent 6196c161e2
commit 99f902d86c
1 changed files with 4 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
melo/download_utils.py
View File

@@ -1,7 +1,7 @@
import torch
import os
from . import utils
from cached_path import cached_path
DOWNLOAD_CKPT_URLS = {
'EN': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/EN/checkpoint.pth',
'EN_V2': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/EN_V2/checkpoint.pth',
@@ -25,23 +25,11 @@ DOWNLOAD_CONFIG_URLS = {
def load_or_download_config(locale):
language = locale.split('-')[0].upper()
assert language in DOWNLOAD_CONFIG_URLS
config_path = os.path.expanduser(f'~/.local/share/openvoice/basespeakers/{language}/config.json')
try:
return utils.get_hparams_from_file(config_path)
except:
# download
os.makedirs(os.path.dirname(config_path), exist_ok=True)
os.system(f'wget {DOWNLOAD_CONFIG_URLS[language]} -O {config_path}')
config_path = cached_path(DOWNLOAD_CONFIG_URLS[language])
return utils.get_hparams_from_file(config_path)
def load_or_download_model(locale, device):
language = locale.split('-')[0].upper()
assert language in DOWNLOAD_CKPT_URLS
ckpt_path = os.path.expanduser(f'~/.local/share/openvoice/basespeakers/{language}/checkpoint.pth')
try:
return torch.load(ckpt_path, map_location=device)
except:
# download
os.makedirs(os.path.dirname(ckpt_path), exist_ok=True)
os.system(f'wget {DOWNLOAD_CKPT_URLS[language]} -O {ckpt_path}')
return torch.load(ckpt_path, map_location=device)
ckpt_path = cached_path(DOWNLOAD_CKPT_URLS[language])
return torch.load(ckpt_path, map_location=device)