blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-10-09 00:10:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

修正删除模型会报错的bug

Browse Source
This commit is contained in:
XTer 2024-03-12 02:03:09 +08:00
parent fdc2df2170
commit 8eb3c556f9
3 changed files with 23 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
GPT_SoVITS/TTS_infer_pack/TTS.py
View File

@ -51,15 +51,19 @@ custom:
class TTS_Config: class TTS_Config:
def __init__(self, configs: Union[dict, str]): def __init__(self, configs: Union[dict, str]):
configs_base_path:str = "GPT_SoVITS/configs/" if isinstance(configs, str) and configs=="":
os.makedirs(configs_base_path, exist_ok=True) self.default_configs:dict = None
self.configs_path:str = os.path.join(configs_base_path, "tts_infer.yaml") self.configs_path = "GPT_SoVITS/configs/tts_infer.yaml"
if isinstance(configs, str): else:
self.configs_path = configs configs_base_path:str = "GPT_SoVITS/configs/"
configs:dict = self._load_configs(configs) os.makedirs(configs_base_path, exist_ok=True)
self.configs_path:str = os.path.join(configs_base_path, "tts_infer.yaml")
if isinstance(configs, str):
self.configs_path = configs
configs:dict = self._load_configs(configs)
# assert isinstance(configs, dict) # assert isinstance(configs, dict)
self.default_configs:dict = configs.get("default", None) self.default_configs:dict = configs.get("default", None)
if self.default_configs is None: if self.default_configs is None:
self.default_configs={ self.default_configs={
"device": "cpu", "device": "cpu",
@ -70,7 +74,10 @@ class TTS_Config:
"bert_base_path": "GPT_SoVITS/pretrained_models/chinese-roberta-wwm-ext-large", "bert_base_path": "GPT_SoVITS/pretrained_models/chinese-roberta-wwm-ext-large",
"flash_attn_enabled": True "flash_attn_enabled": True
} }
self.configs:dict = configs.get("custom", self.default_configs) if isinstance(configs, dict):
self.configs:dict = configs.get("custom", self.default_configs)
else:
self.configs:dict = self.default_configs
self.device = self.configs.get("device") self.device = self.configs.get("device")
self.is_half = self.configs.get("is_half") self.is_half = self.configs.get("is_half")

2
Inference

@ -1 +1 @@
Subproject commit 6db34d726ae2d84ab7982865343bf966246e00d4 Subproject commit 03326be908f0288ab8d26f435eca2ca1ec950b90

8
trained/character_info.json
View File

@ -1,4 +1,8 @@
{ {
"deflaut_character": "胡桃(测试)" "deflaut_character": "胡桃(测试)",
"characters_and_emotions": {
"胡桃(测试)": [
"default"
]
}
} }