torch == 2.2.0
transformers == 4.42.4
einops
bitsandbytes
accelerate