Optimum Transformers
Accelerated NLP pipelines for fast inference
Installation:
With PyPI:
pip install optimum-transformers
Or directly from GitHub:
pip install git+https://github.com/AlekseyKorshuk/optimum-transformers
Usage:
The pipeline API is similar to transformers pipeline with just a few differences which are explained below.
Just provide the path/url to the model, and it'll download the model if needed from the hub and automatically create onnx graph and run inference.
from optimum_transformers import pipeline
# Initialize a pipeline by passing the task name and
# set onnx to True (default value is also True)
nlp = pipeline("sentiment-analysis", use_onnx=True)
nlp("Transformers and onnx runtime is an awesome combo!")
# [{'label': 'POSITIVE', 'score': 0.999721109867096}]
Or provide a different model using the model argument.
from optimum_transformers import pipeline
nlp = pipeline("question-answering", model="deepset/roberta-base-squad2", use_onnx=True)
nlp(question="What is ONNX Runtime ?",
context="ONNX Runtime is a highly performant single inference engine for multiple platforms and hardware")
# {'answer': 'highly performant single inference engine for multiple platforms and hardware', 'end': 94,
# 'score': 0.751201868057251, 'start': 18}
from optimum_transformers import pipeline
nlp = pipeline("ner", model="mys/electra-base-turkish-cased-ner", use_onnx=True, optimize=True,
grouped_entities=True)
nlp("adana kebap ülkemizin önemli lezzetlerinden biridir.")
# [{'entity_group': 'B-food', 'score': 0.869149774312973, 'word': 'adana kebap'}]
Set use_onnx to False for standard torch inference. Set optimize to True for quantize with ONNX. ( set use_onnx to True)
Supported pipelines
You can create Pipeline objects for the following down-stream tasks:
feature-extraction: Generates a tensor representation for the input sequencenerandtoken-classification: Generates named entity mapping for each word in the input sequence.sentiment-analysis: Gives the polarity (positive / negative) of the whole input sequence. Can be used for any text classification model.question-answering: Provided some context and a question referring to the context, it will extract the answer to the question in the context.text-classification: Classifies sequences according to a given number of classes from training.zero-shot-classification: Classifies sequences according to a given number of classes directly in runtime.fill-mask: The task of masking tokens in a sequence with a masking token, and prompting the model to fill that mask with an appropriate token.text-generation: The task of generating text according to the previous text provided.
Calling the pipeline for the first time loads the model, creates the onnx graph, and caches it for future use. Due to this, the first load will take some time. Subsequent calls to the same model will load the onnx graph automatically from the cache.
Benchmarks
Note: For some reason, onnx is slow on colab notebook, so you won't notice any speed-up there. Benchmark it on your own hardware.
Check our example of benchmarking: example.
For detailed benchmarks and other information refer to this blog post and notebook.
- Accelerate your NLP pipelines using Hugging Face Transformers and ONNX Runtime
- Exporting
🤗 transformers model to ONNX
Note: These results were collected on my local machine. So if you have high performance machine to benchmark, please contact me.
Benchmark sentiment-analysis pipeline
Benchmark zero-shot-classification pipeline
Benchmark token-classification pipeline
Benchmark question-answering pipeline
Benchmark fill-mask pipeline
About
Built by Aleksey Korshuk
Star this repository:
Resources
- Inspired by Huggingface Infinity
- First step done by Suraj Patil
- Optimum
- ONNX
8k Jan 4, 2023
16 Dec 22, 2022
![A repository that shares tuning results of trained models generated by TensorFlow / Keras. Post-training quantization (Weight Quantization, Integer Quantization, Full Integer Quantization, Float16 Quantization), Quantization-aware training. TensorFlow Lite. OpenVINO. CoreML. TensorFlow.js. TF-TRT. MediaPipe. ONNX. [.tflite,.h5,.pb,saved_model,tfjs,tftrt,mlmodel,.xml/.bin, .onnx]](https://user-images.githubusercontent.com/33194443/104581604-2592cb00-56a2-11eb-9610-5eaa0afb6e1f.png)
18 Nov 6, 2022
411 Dec 16, 2022
21 Oct 18, 2022
5 Nov 21, 2022
69 Sep 10, 2022
1 Dec 10, 2021
13 Dec 11, 2022
5.9k Dec 31, 2022
136 Dec 21, 2022
16 Sep 22, 2022
1 Nov 18, 2021
185 Dec 26, 2022
1 Jul 21, 2022
1.4k Jan 04, 2023
87 Dec 19, 2022
7.8k Jan 03, 2023
70 Dec 14, 2022
1 Jan 05, 2022
4.2k Jan 01, 2023
1 Feb 09, 2022
23 Oct 21, 2022