pypi.org : sagemaker-huggingface-inference-toolkit : 1.3.1
Open source library for running inference workload with Hugging Face Deep Learning Containers on Amazon SageMaker.
Registry -
Documentation -
Download -
JSON
Integrity: sha256-398f82d429e6b0a17a71... -
purl: pkg:pypi/[email protected]
Published:
Indexed:
Loading...
Readme
Loading...
Dependencies
- boto3 * benchmark
- locust * benchmark
- multi-model-server >=1.1.4 dev
- retrying * dev
- tensorflow >=2.4.0 dev
- torch >=1.8.0 dev
- transformers[sentencepiece,sklearn] >=4.5.1 dev
- multi-model-server >=1.1.4 mms
- retrying * mms
- black ==21.4b0 quality
- flake8 >=3.8.3 quality
- isort >=5.5.4 quality
- huggingface-hub >=0.0.8
- numpy *
- retrying *
- sagemaker-inference >=1.5.11
- tensorflow >=2.4.0 tensorflow
- black ==21.4b0 test
- boto3 * test
- datasets * test
- mock ==2.0.0 test
- parameterized * test
- psutil * test
- pytest * test
- pytest-sugar * test
- pytest-xdist * test
- sagemaker * test
- torch >=1.8.0 torch
- transformers[sentencepiece,sklearn] >=4.5.1 transformers