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

Merge pull request #263 from HideLord/main

Browse Source 
Fixing compatibility with GPTQ repository
This commit is contained in:
oobabooga 2023-03-12 12:42:08 -03:00 committed by GitHub
commit 901dcba9b4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/quantized_LLaMA.py
View File

@ -41,7 +41,7 @@ def load_quantized_LLaMA(model_name):
print(f"Could not find {pt_model}, exiting...") print(f"Could not find {pt_model}, exiting...")
exit() exit()
model = load_quant(path_to_model, pt_path, bits) model = load_quant(path_to_model, os.path.abspath(pt_path), bits)
# Multi-GPU setup # Multi-GPU setup
if shared.args.gpu_memory: if shared.args.gpu_memory: