accelerate
bitsandbytes
torch
transformers==4.43.1
einops
sentencepiece