
    Iui                     ^    d dl Z d dlZd dlZd dlmZ  G d de      Zd Zedk(  r e        yy)    N)MarianConverterc                   (     e Zd ZdZdef fdZ xZS )OpusMTConverterz%Converts models trained with OPUS-MT.	model_dirc                    t        t        j                  j                  |d      d      5 }t	        j
                  |      }ddd       t        j                  j                  |d   d         }|d   D cg c]"  }t        j                  j                  ||      $ }}t        |   ||       y# 1 sw Y   pxY wc c}w )zvInitializes the OPUS-MT converter.

        Arguments:
          model_dir: Path the OPUS-MT model directory.
        zdecoder.ymlzutf-8)encodingNmodelsr   vocabs)openospathjoinyaml	safe_loadsuper__init__)selfr   decoder_filedecoder_config
model_pathr   vocab_paths	__class__s          P/opt/pipecat/venv/lib/python3.12/site-packages/ctranslate2/converters/opus_mt.pyr   zOpusMTConverter.__init__   s     GGLLM2W
 	:!^^L9N	:
 WW\\)^H-Ea-HI
6DX6N
.2BGGLLD)
 
 	[1	: 	:
s   B28'B>2B;)__name__
__module____qualname____doc__strr   __classcell__)r   s   @r   r   r   	   s    /2# 2 2    r   c                  
   t        j                  t         j                        } | j                  ddd       t        j                  |        | j                         }t	        |j                        }|j                  |       y )N)formatter_classz--model_dirTz$Path to the OPUS-MT model directory.)requiredhelp)	argparseArgumentParserArgumentDefaultsHelpFormatteradd_argumentr   declare_arguments
parse_argsr   convert_from_args)parserargs	converters      r   mainr/      sr    $$ >>F +Q   %%f-D/I%r    __main__)r%   r   r   ctranslate2.converters.marianr   r   r/   r    r    r   <module>r3      s6     	  92o 2*
& zF r    