Spaces:
Running
Running
Fix cache dir bug
Browse files- svs_utils.py +2 -3
svs_utils.py
CHANGED
|
@@ -14,7 +14,6 @@ from datasets import load_dataset
|
|
| 14 |
# "Model②(Multilingual)-jp": "espnet/mixdata_svs_visinger2_spkembed_lang_pretrained",
|
| 15 |
# }
|
| 16 |
|
| 17 |
-
espnet_downloader = ModelDownloader()
|
| 18 |
|
| 19 |
singer_embeddings = {
|
| 20 |
"espnet/aceopencpop_svs_visinger2_40singer_pretrain": {
|
|
@@ -51,8 +50,8 @@ def svs_warmup(config):
|
|
| 51 |
Return: the inference prototype function (which creates pitch/duration and runs model-specific inference)
|
| 52 |
"""
|
| 53 |
if config.model_path.startswith("espnet"):
|
| 54 |
-
|
| 55 |
-
|
| 56 |
model = SingingGenerate(
|
| 57 |
train_config=downloaded["train_config"],
|
| 58 |
model_file=downloaded["model_file"],
|
|
|
|
| 14 |
# "Model②(Multilingual)-jp": "espnet/mixdata_svs_visinger2_spkembed_lang_pretrained",
|
| 15 |
# }
|
| 16 |
|
|
|
|
| 17 |
|
| 18 |
singer_embeddings = {
|
| 19 |
"espnet/aceopencpop_svs_visinger2_40singer_pretrain": {
|
|
|
|
| 50 |
Return: the inference prototype function (which creates pitch/duration and runs model-specific inference)
|
| 51 |
"""
|
| 52 |
if config.model_path.startswith("espnet"):
|
| 53 |
+
espnet_downloader = ModelDownloader(config.cache_dir)
|
| 54 |
+
downloaded = espnet_downloader.download_and_unpack(config.model_path)
|
| 55 |
model = SingingGenerate(
|
| 56 |
train_config=downloaded["train_config"],
|
| 57 |
model_file=downloaded["model_file"],
|