Devsyril/opus-mt-fr-en-onnx

Export ONNX du modèle Helsinki-NLP/opus-mt-fr-en (Helsinki-NLP / OPUS-MT), généré avec 🤗 Optimum.

Contient les composants ONNX standards pour un modèle seq2seq : encoder_model.onnx, decoder_model.onnx, decoder_with_past_model.onnx (selon la version d'Optimum utilisée), ainsi que le tokenizer d'origine.

Utilisation

from optimum.onnxruntime import ORTModelForSeq2SeqLM
from transformers import AutoTokenizer

repo_id = "Devsyril/opus-mt-fr-en-onnx"
model = ORTModelForSeq2SeqLM.from_pretrained(repo_id)
tokenizer = AutoTokenizer.from_pretrained(repo_id)

inputs = tokenizer("Ton texte ici", return_tensors="pt")
generated = model.generate(**inputs)
print(tokenizer.batch_decode(generated, skip_special_tokens=True))

Modèle source : Helsinki-NLP/opus-mt-fr-en