torch == 2.2.0
transformers
einops
bitsandbytes
accelerate