torch
torchvision
transformers
accelerate
bitsandbytes