transformers
torch==2.2.0
accelerate
flash-attn==2.2.0