transformers
torch
torchvision
shortuuid
sentencepiece
protobuf
google
accelerate
flash-attn