2023-07-05 10:33:12 +08:00
|
|
|
accelerate>=0.19.0
|
2023-07-06 12:14:23 +08:00
|
|
|
boto3
|
|
|
|
|
colossalai
|
|
|
|
|
cpm_kernels
|
2023-07-05 10:33:12 +08:00
|
|
|
datasets>=2.12.0
|
|
|
|
|
evaluate>=0.3.0
|
2023-07-06 12:14:23 +08:00
|
|
|
fairscale
|
2023-07-05 10:33:12 +08:00
|
|
|
faiss_gpu==1.7.2
|
2023-07-06 12:14:23 +08:00
|
|
|
jieba
|
|
|
|
|
mmengine>0.8.0
|
2023-07-05 10:33:12 +08:00
|
|
|
nltk==3.8
|
|
|
|
|
numpy==1.23.4
|
2023-07-06 12:14:23 +08:00
|
|
|
openai
|
|
|
|
|
pandas<2.0.0
|
2023-07-05 10:33:12 +08:00
|
|
|
rank_bm25==0.2.2
|
|
|
|
|
requests==2.28.1
|
|
|
|
|
scikit_learn==1.2.1
|
|
|
|
|
sentence_transformers==2.2.2
|
2023-07-06 12:14:23 +08:00
|
|
|
tabulate
|
|
|
|
|
tiktoken
|
|
|
|
|
tokenizers>=0.13.3
|
2023-07-05 10:33:12 +08:00
|
|
|
torch>=1.13.1
|
|
|
|
|
tqdm==4.64.1
|
|
|
|
|
transformers>=4.29.1
|
