|
--- |
|
language: |
|
- en |
|
- fr |
|
- ro |
|
- de |
|
datasets: |
|
- c4 |
|
tags: |
|
- summarization |
|
- translation |
|
|
|
license: apache-2.0 |
|
--- |
|
|
|
# ONNX convert of t5-small |
|
|
|
## Conversion of [t5-small](https://huggingface.co/t5-small) |
|
|
|
## Model description |
|
|
|
More information needed |
|
|
|
## Intended uses & limitations |
|
|
|
You can use this model with Transformers *pipeline*. |
|
|
|
```python |
|
from transformers import AutoTokenizer, pipeline |
|
from optimum.onnxruntime import ORTModelForSeq2SeqLM |
|
tokenizer = AutoTokenizer.from_pretrained("optimum/t5-small") |
|
model = ORTModelForSeq2SeqLM.from_pretrained("optimum/t5-small") |
|
translator = pipeline("translation_en_to_de", model=model, tokenizer=tokenizer) |
|
example = "My name is Wolfgang and I live in Berlin" |
|
results = translator(example) |
|
print(results) |
|
``` |
|
|
