Pipeline: translation_xx_to_yy not working for Mbart

I’m working on a translation tool using Mbart, and I’m trying to translate some text. I’m using pipeline(ā€œtranslation_de_to_enā€, model=model, tokenizer=tokenizer)
which should work, and is how the documentation describes. The error I’m getting is ā€œKeyError: ā€˜deā€™ā€, which I have found out that its the ā€˜de’ in translation_de_to_en.

Anything would be appreciated :upside_down_face:

I’m using facebook/mbart-large-cc25 btw

were you able to figure out the issue? I’m stuck in a similar situation.