transformers==4.41.2
accelerate==0.31.0
flash-attention==1.0.0