datasets==2.19.1
numpy==1.26.4
pandas==2.2.2
faiss-cpu==1.8.0.post1
faiss-gpu==1.7.2
torch==2.3.0
torchmetrics==1.4.0.post0
tqdm==4.66.4
transformers==4.41.0
wandb==0.17.0
xformers==0.0.26.post1