requirements.txt

accelerate>=0.19.0
boto3
colossalai
cpm_kernels
datasets>=2.12.0
evaluate>=0.3.0
fairscale
faiss_gpu==1.7.2
jieba
mmengine>0.8.0
nltk==3.8
numpy==1.23.4
openai
pandas<2.0.0
rank_bm25==0.2.2
requests==2.28.1
scikit_learn==1.2.1
sentence_transformers==2.2.2
tabulate
tiktoken
tokenizers>=0.13.3
torch>=1.13.1
tqdm==4.64.1
transformers>=4.29.1