CLEX-Chat / requirements.txt
Guanzheng's picture
Update requirements.txt
b908c73
raw
history blame
116 Bytes
torch==2.0.1
accelerate==0.23.0
tqdm==4.64.1
torchdiffeq==0.2.3
transformers==4.34.0
flash-attn==2.3.2
PyPDF2
einops