accelerate
timm
einops
torch
torchvision
git+https://github.com/huggingface/transformers.git
sentencepiece