torch>=1.9.0
torchvision
regex
ftfy
einops
pandas
braceexpand
fsspec
tqdm
huggingface_hub
sentencepiece
protobuf<4
timm
transformers
webdataset
pyarrow
pytest-split==0.8.0
pytest==7.2.0