Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-12-25 13:58:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

Install exllama (#83)

Browse Source 
* Install exllama

* Handle updating exllama
This commit is contained in:
jllllll 2023-06-17 17:10:36 -05:00 committed by GitHub
parent 657049d7d0
commit b1d05cbbf6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
webui.py
View File

@ -167,8 +167,17 @@ def update_dependencies():
if not os.path.exists("repositories/"): if not os.path.exists("repositories/"):
os.mkdir("repositories") os.mkdir("repositories")
# Install GPTQ-for-LLaMa which enables 4bit CUDA quantization
os.chdir("repositories") os.chdir("repositories")
# Install or update exllama as needed
if not os.path.exists("exllama/"):
run_cmd("git clone https://github.com/turboderp/exllama.git", environment=True)
else:
os.chdir("exllama")
run_cmd("git pull", environment=True)
os.chdir("..")
# Install GPTQ-for-LLaMa which enables 4bit CUDA quantization
if not os.path.exists("GPTQ-for-LLaMa/"): if not os.path.exists("GPTQ-for-LLaMa/"):
run_cmd("git clone https://github.com/oobabooga/GPTQ-for-LLaMa.git -b cuda", assert_success=True, environment=True) run_cmd("git clone https://github.com/oobabooga/GPTQ-for-LLaMa.git -b cuda", assert_success=True, environment=True)