rename

2024-02-19 23:15:47 +00:00
parent 734228934f
commit db237ce6a5
62 changed files with 14 additions and 17 deletions
--- a/melo/download_utils.py
+++ b/melo/download_utils.py
@@ -0,0 +1,47 @@
+import torch
+import os
+from . import utils
+
+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',
+    'FR': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/FR/checkpoint.pth',
+    'JP': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/JP/checkpoint.pth',
+    'ES': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/ES/checkpoint.pth',
+    'ZH': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/ZH/checkpoint.pth',
+    'KR': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/KR/checkpoint.pth',
+}
+
+DOWNLOAD_CONFIG_URLS = {
+    'EN': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/EN/config.json',
+    'EN_V2': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/EN_V2/config.json',
+    'FR': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/FR/config.json',
+    'JP': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/JP/config.json',
+    'ES': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/ES/config.json',
+    'ZH': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/ZH/config.json',
+    'KR': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/KR/config.json',
+}
+
+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}')
+    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)