OPUS-MT

OPUS-MT contains a collection of 1000+ pretrained models. Since the models are Transformer models trained with Marian, they are compatible with CTranslate2.

The Marian guide also applies for these models, but a separate converter is provided for convenience:

ct2-opus-mt-converter --model_dir opus_model --output_dir ct2_model 

Example

This example uses the English-German model:

wget https://object.pouta.csc.fi/OPUS-MT-models/en-de/opus-2020-02-26.zip unzip opus-2020-02-26.zip ct2-opus-mt-converter --model_dir . --output_dir ende_ctranslate2 

import ctranslate2 import sentencepiece as spm sp = spm.SentencePieceProcessor() sp.load("source.spm") source = sp.encode("Hello world!", out_type=str) translator = ctranslate2.Translator("ende_ctranslate2") results = translator.translate_batch([source]) output = sp.decode(results[0].hypotheses[0]) print(output)