README.md

metadata

language:
  - en
  - fr
  - ro
  - de
datasets:
  - c4
tags:
  - summarization
  - translation
license: apache-2.0

ONNX convert of t5-small

Conversion of t5-small

Model description

More information needed

Intended uses & limitations

You can use this model with Transformers pipeline.

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)