Helsinki-NLP/opus-mt-tc-big-en-cat_oci_spa · Update checkpoint for transformers>=4.29

Language Technology Research Group at the University of Helsinki org Oct 10, 2023

Automatically merging the PR.

ArthurZ changed pull request status to merged Oct 10, 2023

Language Technology Research Group at the University of Helsinki org Oct 10, 2023

Following the merge of a PR in transformers it appeared that this model was not properly converted. This PR will fix the inference and was tested using the following script:

>>> from transformers import AutoTokenizer, MarianMTModel
>>> tokenizer = AutoTokenizer.from_pretrained('Helsinki-NLP/opus-mt-tc-big-en-cat_oci_spa')
>>> inputs = tokenizer("Hey! Let's learn together", return_tensors="pt", padding=True)
>>> model = MarianMTModel.from_pretrained('Helsinki-NLP/opus-mt-tc-big-en-cat_oci_spa')
>>> print(tokenizer.batch_decode(model.generate(**inputs)))
['<pad> ¡Oigan! Aprendamos juntos</s>']

Update checkpoint for transformers>=4.29a3156af6