Merge pull request #78 from elvizlai/main

fix: users.pth path check
2 years ago · 0aae34a321
parent e63672943a e0590b9c26
commit 0aae34a321
1 changed files with 6 additions and 5 deletions
--- a/webui.py
+++ b/webui.py
@ -20,6 +20,7 @@ if(site_packages_roots==[]):site_packages_roots=["%s/runtime/Lib/site-packages"
 #os.environ["OPENBLAS_NUM_THREADS"] = "4"
 os.environ["no_proxy"] = "localhost, 127.0.0.1, ::1"
 for site_packages_root in site_packages_roots:
+    if os.path.exists("%s/users.pth" % (site_packages_root)):
        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"