SageMaker Hugging Face Inference Toolkit is an open-source library for serving Transformers models on Amazon SageMaker. This library provides default pre-processing, predict and postprocessing for certain Transformers models and tasks. It utilizes the SageMaker Inference Toolkit for starting up the model server, which is responsible for handling inference requests. For the Dockerfiles used for building SageMaker Hugging Face Containers, see AWS Deep Learning Containers. The SageMaker Hugging Face Inference Toolkit implements various additional environment variables to simplify your deployment experience. The Hugging Face Inference Toolkit allows user to override the default methods of the HuggingFaceHandlerService. SageMaker Hugging Face Inference Toolkit is licensed under the Apache 2.0 License.

Features

  • Create a Amazon SageMaker endpoint with a model from the Hub
  • Create a Amazon SageMaker endpoint with a trained model
  • The HF_TASK environment variable defines the task for the used Transformers pipeline
  • The HF_MODEL_ID environment variable defines the model id, which will be automatically loaded
  • The HF_API_TOKEN environment variable defines the your Hugging Face authorization token
  • User defined code/modules

Project Samples

Project Activity

See All Activity >

License

Apache License V2.0

Follow SageMaker Hugging Face Inference Toolkit

SageMaker Hugging Face Inference Toolkit Web Site

You Might Also Like
Control remote support software for remote workers and IT teams Icon
Control remote support software for remote workers and IT teams

Raise the bar for remote support and reduce customer downtime.

ConnectWise ScreenConnect, formerly ConnectWise Control, is a remote support solution for Managed Service Providers (MSP), Value Added Resellers (VAR), internal IT teams, and managed security providers. Fast, reliable, secure, and simple to use, ConnectWise ScreenConnect helps businesses solve their customers' issues faster from any location. The platform features remote support, remote access, remote meeting, customization, and integrations with leading business tools.
Rate This Project
Login To Rate This Project

User Reviews

Be the first to post a review of SageMaker Hugging Face Inference Toolkit!

Additional Project Details

Programming Language

Python

Related Categories

Python UML Tool, Python Libraries, Python Deep Learning Frameworks

Registered

2022-07-07