torch==2.5.1
torchvision==0.20.1
scipy
einops
transformers==4.51.1
datasets
psutil
peft
accelerate
scikit-learn
timm
matplotlib
seaborn

# winoground
tensorflow

# needs to be separately installed after other dependencies are present
# pip install flash_attn
