oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2025-12-06 07:12:10 +01:00
Code Issues Actions 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():
site_packages_path = None
for sitedir in site.getsitepackages():
if "site-packages" in sitedir and conda_env_path in sitedir:
site_packages_path = sitedir
@ -69,6 +70,7 @@ def torch_version():
def is_installed():
site_packages_path = None
for sitedir in site.getsitepackages():
if "site-packages" in sitedir and conda_env_path in sitedir:
site_packages_path = sitedir