sagemaker-inference>=1.5.5
huggingface_hub>=0.0.8
retrying
numpy

[benchmark]
boto3
locust

[dev]
transformers[sentencepiece,sklearn]>=4.5.1
multi-model-server>=1.1.4
retrying
torch>=1.8.0
tensorflow>=2.4.0

[mms]
multi-model-server>=1.1.4
retrying

[quality]
black==21.4b0
isort>=5.5.4
flake8>=3.8.3

[tensorflow]
tensorflow>=2.4.0

[test]
pytest
pytest-xdist
parameterized
psutil
datasets
pytest-sugar
black==21.4b0
sagemaker
boto3
mock==2.0.0

[torch]
torch>=1.8.0

[transformers]
transformers[sentencepiece,sklearn]>=4.5.1
