config.py :

现在在检查半精度计算前会先检查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

View File

@ -29,6 +29,7 @@ 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())

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]
#os.environ["OPENBLAS_NUM_THREADS"] = "4"
os.environ["no_proxy"] = "localhost, 127.0.0.1, ::1"
os.environ["all_proxy"] = ""
for site_packages_root in site_packages_roots:
if os.path.exists(site_packages_root):
with open("%s/users.pth" % (site_packages_root), "w") as f:
print(site_packages_roots)
try:
with open("%s/users.pth" % (site_packages_root), "w+") as f:
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
import traceback
import shutil