torch
accelerate
transformers
bitsandbytes