| import sys,os |
|
|
| import torch |
|
|
| |
| sovits_path = "" |
| gpt_path = "" |
| is_half_str = os.environ.get("is_half", "True") |
| is_half = True if is_half_str.lower() == 'true' else False |
| is_share_str = os.environ.get("is_share","False") |
| is_share= True if is_share_str.lower() == 'true' else False |
|
|
| cnhubert_path = "GPT_SoVITS/pretrained_models/chinese-hubert-base" |
| bert_path = "GPT_SoVITS/pretrained_models/chinese-roberta-wwm-ext-large" |
| pretrained_sovits_path = "GPT_SoVITS/pretrained_models/s2G488k.pth" |
| pretrained_gpt_path = "GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt" |
|
|
| exp_root = "logs" |
| python_exec = sys.executable or "python" |
| if torch.cuda.is_available(): |
| infer_device = "cuda" |
| else: |
| infer_device = "cpu" |
|
|
| webui_port_main = 9874 |
| webui_port_uvr5 = 9873 |
| webui_port_infer_tts = 9872 |
| webui_port_subfix = 9871 |
|
|
| api_port = 9880 |
|
|
| if infer_device == "cuda": |
| gpu_name = torch.cuda.get_device_name(0) |
| if ( |
| ("16" in gpu_name and "V100" not in gpu_name.upper()) |
| or "P40" in gpu_name.upper() |
| or "P10" in gpu_name.upper() |
| or "1060" in gpu_name |
| or "1070" in gpu_name |
| or "1080" in gpu_name |
| ): |
| is_half=False |
|
|
| if(infer_device=="cpu"):is_half=False |
|
|
| class Config: |
| def __init__(self): |
| self.sovits_path = sovits_path |
| self.gpt_path = gpt_path |
| self.is_half = is_half |
|
|
| self.cnhubert_path = cnhubert_path |
| self.bert_path = bert_path |
| self.pretrained_sovits_path = pretrained_sovits_path |
| self.pretrained_gpt_path = pretrained_gpt_path |
|
|
| self.exp_root = exp_root |
| self.python_exec = python_exec |
| self.infer_device = infer_device |
|
|
| self.webui_port_main = webui_port_main |
| self.webui_port_uvr5 = webui_port_uvr5 |
| self.webui_port_infer_tts = webui_port_infer_tts |
| self.webui_port_subfix = webui_port_subfix |
|
|
| self.api_port = api_port |
|
|
