tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Fix: roll back #170 and fix bug in api /tools/random_example (#177)

Browse Source
This commit is contained in:
Sora
2023-11-11 20:28:31 +08:00
committed by GitHub
parent 5c37dbd3cf
commit 9483a6c576
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
server.py
View File

@@ -50,7 +50,7 @@ chrsMap: List[Dict[int, str]] = list()
# 加载模型 # 加载模型
models = config.server_config.models models = config.server_config.models
for model in models: for model in models:
hps_List.append(utils.get_hparams_from_file(model["config_path"])) hps_List.append(utils.get_hparams_from_file(model["config"]))
# 添加角色字典 # 添加角色字典
chrsMap.append(dict()) chrsMap.append(dict())
for name, cid in hps_List[-1].data.spk2id.items(): for name, cid in hps_List[-1].data.spk2id.items():

4
server_fastapi.py
View File

@@ -163,7 +163,7 @@ if __name__ == "__main__":
models_info = config.server_config.models models_info = config.server_config.models
for model_info in models_info: for model_info in models_info:
loaded_models.init_model( loaded_models.init_model(
config_path=model_info["config_path"], config_path=model_info["config"],
model_path=model_info["model"], model_path=model_info["model"],
device=model_info["device"], device=model_info["device"],
language=model_info["language"], language=model_info["language"],
@@ -550,7 +550,7 @@ if __name__ == "__main__":
examples = all_examples[root_dir] examples = all_examples[root_dir]
# 从项目Data目录中搜索train/val.list # 从项目Data目录中搜索train/val.list
for root, directories, _files in os.walk("Data"): for root, directories, _files in os.walk(root_dir):
for file in _files: for file in _files:
if file in ["train.list", "val.list"]: if file in ["train.list", "val.list"]:
with open( with open(