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