accelerate
bitsandbytes
timm
einops
torch
torchvision
Pillow
transformers==4.43.0
tiktoken