pypi.org : sagemaker-huggingface-inference-toolkit : 2.4.1
Open source library for running inference workload with Hugging Face Deep Learning Containers on Amazon SageMaker.
Registry -
Documentation -
Download -
JSON
Integrity: sha256-3a11a7808a7a8e2ddfcb... -
purl: pkg:pypi/[email protected]
Published:
Indexed:
Loading...
Readme
Loading...
Dependencies
- boto3 * extra == "benchmark" (optional)
- locust * extra == "benchmark" (optional)
- diffusers >=0.23.0 extra == "dev" (optional)
- multi-model-server >=1.1.4 extra == "dev" (optional)
- retrying * extra == "dev" (optional)
- tensorflow <2.11,>=2.4.0 extra == "dev" (optional)
- torch >=1.8.0 extra == "dev" (optional)
- torchaudio * extra == "dev" (optional)
- transformers[sentencepiece,sklearn] >=4.17.0 extra == "dev" (optional)
- diffusers >=0.23.0 extra == "diffusers" (optional)
- multi-model-server >=1.1.4 extra == "mms" (optional)
- retrying * extra == "mms" (optional)
- black >=21.10 extra == "quality" (optional)
- flake8 >=3.8.3 extra == "quality" (optional)
- isort >=5.5.4 extra == "quality" (optional)
- tensorflow <2.11,>=2.4.0 extra == "tensorflow" (optional)
- black ==21.4b0 extra == "test" (optional)
- boto3 * extra == "test" (optional)
- datasets * extra == "test" (optional)
- mock ==2.0.0 extra == "test" (optional)
- parameterized * extra == "test" (optional)
- psutil * extra == "test" (optional)
- pytest <8 extra == "test" (optional)
- pytest-sugar * extra == "test" (optional)
- pytest-xdist * extra == "test" (optional)
- sagemaker * extra == "test" (optional)
- torch >=1.8.0 extra == "torch" (optional)
- torchaudio * extra == "torch" (optional)
- transformers[sentencepiece,sklearn] >=4.17.0 extra == "transformers" (optional)
- Pillow *
- huggingface-hub >=0.0.8
- librosa *
- numpy *
- phonemizer *
- pyctcdecode >=0.3.0
- retrying *
- sagemaker-inference >=1.8.0