Location of ctranslate2.converters.OpusMTConverter

tel34 · March 15, 2022, 2:14pm

Hi, Sorry for troubling you but I can’t see the OpusMTConverter in the ctranslate2/converters foilder.
Terence

guillaumekln · March 15, 2022, 2:16pm

Hi,

The file is here:

OpenNMT/CTranslate2/blob/master/python/ctranslate2/converters/opus_mt.py

import argparse
import os

import yaml

from ctranslate2.converters.marian import MarianConverter


class OpusMTConverter(MarianConverter):
    """Converts models trained with OPUS-MT."""

    def __init__(self, model_dir):
        with open(
            os.path.join(model_dir, "decoder.yml"), encoding="utf-8"
        ) as decoder_file:
            decoder_config = yaml.safe_load(decoder_file)

        model_path = os.path.join(model_dir, decoder_config["models"][0])
        vocab_paths = [
            os.path.join(model_dir, path) for path in decoder_config["vocabs"]

This file has been truncated. show original

tel34 · March 15, 2022, 4:22pm

Thank you