# --extra-index-url https://download.pytorch.org/whl/cu118
transformers
# torch
# torchvision
numpy
fire
pillow
numpy
einops
sentencepiece
protobuf