Merge pull request #263 from HideLord/main

Fixing compatibility with GPTQ repository
This commit is contained in:
oobabooga 2023-03-12 12:42:08 -03:00 committed by GitHub
commit 901dcba9b4
WARNING! Although there is a key with this ID in the database it does not verify this commit! This commit is SUSPICIOUS.
GPG key ID: 4AEE18F83AFDEB23

View file

@ -41,7 +41,7 @@ def load_quantized_LLaMA(model_name):
print(f"Could not find {pt_model}, exiting...")
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
if shared.args.gpu_memory: