Project02/AITrain/requirements.txt

20 lines
229 B
Plaintext

openai
tiktoken
tqdm
python-dotenv
torch
transformers>=4.21.0
datasets
scikit-learn
wandb
accelerate
modelscope
vllm
swanlab==0.5.7
accelerate==1.6.0
datasets==3.5.1
peft==0.15.2
autoawq
PyPDF2
sentence-transformersfaiss-cpu
fitz