Update webui.py

This commit is contained in:
RVC-Boss 2024-01-18 17:58:10 +08:00 committed by GitHub
parent 2e500db3e3
commit dbbf616762
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -12,13 +12,19 @@ tmp = os.path.join(now_dir, "TEMP")
os.makedirs(tmp, exist_ok=True)
os.environ["TEMP"] = tmp
import site
site_packages_root="%s/runtime/Lib/site-packages"%now_dir
site_packages_roots = []
for path in site.getsitepackages():
if("site-packages"in path):site_packages_root=path
os.environ["OPENBLAS_NUM_THREADS"] = "4"
if "packages" in path:
site_packages_roots.append(path)
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"
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))
for site_packages_root in site_packages_roots:
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)
)
import traceback
sys.path.append(now_dir)
import shutil