einops
torch==2.0.1
accelerate
bitsandbytes