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

config.py :

Browse Source 
现在在检查半精度计算前会先检查CUDA
webui.py :
  优化users.py的写入
This commit is contained in:
YuChuXi 2024-01-27 14:11:10 +08:00
parent f9387e0af8
commit a0af8c32de
2 changed files with 24 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
config.py
View File

@ -29,16 +29,17 @@ webui_port_subfix = 9871
api_port = 9880 api_port = 9880
gpu_name = torch.cuda.get_device_name(0) if infer_device == "cuda":
if ( gpu_name = torch.cuda.get_device_name(0)
("16" in gpu_name and "V100" not in gpu_name.upper()) if (
or "P40" in gpu_name.upper() ("16" in gpu_name and "V100" not in gpu_name.upper())
or "P10" in gpu_name.upper() or "P40" in gpu_name.upper()
or "1060" in gpu_name or "P10" in gpu_name.upper()
or "1070" in gpu_name or "1060" in gpu_name
or "1080" in gpu_name or "1070" in gpu_name
): or "1080" in gpu_name
is_half=False ):
is_half=False
if(infer_device=="cpu"):is_half=False if(infer_device=="cpu"):is_half=False

18
webui.py
View File

@ -25,13 +25,21 @@ for path in site.getsitepackages():
if(site_packages_roots==[]):site_packages_roots=["%s/runtime/Lib/site-packages" % now_dir] if(site_packages_roots==[]):site_packages_roots=["%s/runtime/Lib/site-packages" % now_dir]
#os.environ["OPENBLAS_NUM_THREADS"] = "4" #os.environ["OPENBLAS_NUM_THREADS"] = "4"
os.environ["no_proxy"] = "localhost, 127.0.0.1, ::1" os.environ["no_proxy"] = "localhost, 127.0.0.1, ::1"
os.environ["all_proxy"] = ""
for site_packages_root in site_packages_roots: for site_packages_root in site_packages_roots:
if os.path.exists(site_packages_root): if os.path.exists(site_packages_root):
with open("%s/users.pth" % (site_packages_root), "w") as f: print(site_packages_roots)
f.write( try:
"%s\n%s/tools\n%s/tools/damo_asr\n%s/GPT_SoVITS\n%s/tools/uvr5" with open("%s/users.pth" % (site_packages_root), "w+") as f:
% (now_dir, now_dir, now_dir, now_dir, now_dir) f.write(
) "%s\n%s/tools\n%s/tools/damo_asr\n%s/GPT_SoVITS\n%s/tools/uvr5"
% (now_dir, now_dir, now_dir, now_dir, now_dir)
)
break
except PermissionError:
pass
from tools import my_utils from tools import my_utils
import traceback import traceback
import shutil import shutil