To use the "translation"
task pipeline
, the tgt_lang
kwarg is set to a single language:
from transformers import pipeline
mt_en_fr = pipeline("translation", model="facebook/m2m100_418M",
tokenizer="facebook/m2m100_418M",
src_lang="en", tgt_lang="fr")
mt_en_de = pipeline("translation", model="facebook/m2m100_418M",
tokenizer="facebook/m2m100_418M",
src_lang="en", tgt_lang="de")
out_en_fr = mt_en_fr("Hello, how are you?", max_length=50)
out_en_de = mt_en_de("Hello, how are you?", max_length=50)
Is there an option for multiple target language in translation pipeline? E.g.
mt = pipeline("translation", model="facebook/m2m100_418M",
tokenizer="facebook/m2m100_418M",
src_lang="en", tgt_langs=["fr", "de"])
mt("Hello, how are you?", max_length=50)