Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-26 01:30:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update one_click.py to initialize site_packages_path variable (#4118)

Browse Source
This commit is contained in:
Nathan Thomas 2023-09-28 19:31:29 +08:00 committed by GitHub
parent 3fb1e0236a
commit e145d9a0da
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
one_click.py
View File

@ -55,6 +55,7 @@ def cpu_has_avx2():
def torch_version(): def torch_version():
site_packages_path = None
for sitedir in site.getsitepackages(): for sitedir in site.getsitepackages():
if "site-packages" in sitedir and conda_env_path in sitedir: if "site-packages" in sitedir and conda_env_path in sitedir:
site_packages_path = sitedir site_packages_path = sitedir
@ -69,6 +70,7 @@ def torch_version():
def is_installed(): def is_installed():
site_packages_path = None
for sitedir in site.getsitepackages(): for sitedir in site.getsitepackages():
if "site-packages" in sitedir and conda_env_path in sitedir: if "site-packages" in sitedir and conda_env_path in sitedir:
site_packages_path = sitedir site_packages_path = sitedir