requirements.txt

# int8
bitsandbytes==0.37.1
accelerate==0.17.1

# chatglm
protobuf>=3.19.5,<3.20.1
transformers==4.27.1
icetk
cpm_kernels==1.0.11

# 
datasets==2.10.1
git+https://github.com/huggingface/peft.git  # 最新版本 >=0.3.0.dev0

-f https://download.pytorch.org/whl/cpu
torch
-f https://download.pytorch.org/whl/cpu
torchvision
-f https://download.pytorch.org/whl/cpu
torchaudio