torch==2.2.0
transformers
flash-attn==2.2.0
einops