torch==2.3.1
accelerate==0.31.0
transformers==4.41.2
bitsandbytes
trl
einops