BERT With Zero Overhead
Run BERT on remote machines with 1ms overhead.
In this tutorial we're going to demonstrate how you can deploy one of the fastest NLP models, BERT, to remote GPUs and maintain decent latency.
BERT encoder is extremely fast, running at 1.5ms on local GPU (tested on Nvidia T4). Deploying that model to remote machines and maintaining low latency is hard.
Everinfer is highly optimized and will allow you to run that model on remote machines while keeping up with its speed.
Install Everinfer and HuggingFace transformers library.
Convert the model to ONNX format:
!python3 -m transformers.onnx --model=distilbert-base-uncased onnx/
Authenticate on Everinfer using your API key, upload the model, and create inference engine:
from everinfer import Client
client = Client('my_api_key')
pipeline = client.register_pipeline('bert', ['onnx/model.onnx'])
runner = client.create_engine(pipeline['uuid'])
You are ready to go!
Since HuggingFace tokenizers are fully compatible with Everinfer expected input format, you can feed tokenizer outputs directly to the deployed model:
from transformers import AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
inputs = tokenizer("Everinfer is fast af", return_tensors="np")
After applying tokenizer to input text, running the model is as simple as:
preds = runner.predict([inputs])
Remote GPU access overhead is virtually zero!
Deploy that code to AWS Lambda to go fully serverless, or use it as a part of your web app.