torch>=1.9.0
transformers>=4.20.0
tqdm>=4.50.0
scikit-learn>=0.24.0
pandas>=1.2.0
numpy>=1.21.0
datasets>=2.0.0
sentence-transformers>=2.0.0
faiss-cpu>=1.7.0

[cuda]
flash-attn>=2.5.0

[harness]
lm-eval==0.4.8
