torch==2.1.2
transformers==4.37.2
flash-attn==2.4.2