
    C1i                     `    S SK r S SKrS SKrS SKJr   " S S\5      rS r\S:X  a  \" 5         gg)    N)MarianConverterc                   4   ^  \ rS rSrSrS\4U 4S jjrSrU =r$ )OpusMTConverter	   z%Converts models trained with OPUS-MT.	model_dirc                   > [        [        R                  R                  US5      SS9 n[        R
                  " U5      nSSS5        [        R                  R                  UWS   S   5      nUS    Vs/ s H"  n[        R                  R                  X5      PM$     nn[        TU ]  XF5        g! , (       d  f       Nv= fs  snf )z^Initializes 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          X/home/james-whalen/.local/lib/python3.13/site-packages/ctranslate2/converters/opus_mt.pyr   OpusMTConverter.__init__   s     GGLLM2W
!^^L9N

 WW\\)^H-Ea-HI
6DX6N
6NdBGGLL)6N 	 
 	1
 

s   B16)C1
B? )	__name__
__module____qualname____firstlineno____doc__strr   __static_attributes____classcell__)r   s   @r   r   r   	   s    /2# 2 2    r   c                      [         R                  " [         R                  S9n U R                  SSSS9  [        R                  U 5        U R                  5       n[	        UR                  5      nUR                  U5        g )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   mainr4      sr    $$ >>F +Q   %%f-D/I%r%   __main__)r*   r   r   ctranslate2.converters.marianr   r   r4   r   r   r%   r   <module>r7      s6     	  92o 2*
& zF r%   