torch
torchvision
git+https://github.com/Dao-AILab/flash-attention.git
pip==24.2
ninja
setuptools
wheel