Merge pull request #34 from winglian/v2-fixes

fixes for most recent update
This commit is contained in:
John Smith 2023-03-28 23:49:56 +08:00 committed by GitHub
commit b5e3dae573
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 1 deletions

View File

@ -1,6 +1,6 @@
import torch
import numpy as np
import quant_cuda
from gptq_llama import quant_cuda
# Global Buffer

View File

@ -3,6 +3,7 @@ accelerate
bitsandbytes
datasets
sentencepiece
safetensors
git+https://github.com/huggingface/transformers.git
git+https://github.com/sterlind/GPTQ-for-LLaMa.git@lora_4bit
git+https://github.com/sterlind/peft.git