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

Fix possible crash (#53)

Browse Source
This commit is contained in:
Semjon Kravtšenko 2023-05-06 07:14:09 +03:00 committed by GitHub
parent dec31af910
commit 126d216384
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
webui.py
View File

@ -130,7 +130,8 @@ def update_dependencies():
run_cmd("conda install -y -k gxx_linux-64=11.2.0", environment=True)
# Compile and install GPTQ-for-LLaMa
os.rename("setup_cuda.py", "setup.py")
if os.path.exists('setup_cuda.py'):
os.rename("setup_cuda.py", "setup.py")
run_cmd("python -m pip install .", environment=True)
# Wheel installation can fail while in the build directory of a package with the same name