:j jddlZddlZddlZddlZddlmZddlmZmZGddejZ y)N)Optional)ACCEPTED_MODEL_TYPES ModelSpecc eZdZdZedej dej fdZdejde fdZ dde d e e d e e d e de f d Z ejd Zy) Converterz Base class for model converters.parserreturnc|jddd|jddd|jd dtd |jd d d|S)zAdds common conversion options to the command line parser. Arguments: parser: Command line argument parser. z --output_dirTzOutput model directory.)requiredhelpz--vocab_mappingNz#Vocabulary mapping file (optional).)defaultr z--quantizationzWeight quantization type.)r choicesr z--force store_truez=Force conversion even if the output directory already exists.)actionr ) add_argumentr)rs u/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/ctranslate2/converters/converter.pydeclare_argumentszConverter.declare_argumentss  T0I    t2W    (,   P   argsc||j|j|j|j|jS)aHelper function to call :meth:`ctranslate2.converters.Converter.convert` with the parsed command line options. Arguments: args: Namespace containing parsed arguments. Returns: Path to the output directory. )vmap quantizationforce)convert output_dir vocab_mappingrr)selfrs rconvert_from_argszConverter.convert_from_args(s:|| OO##****   rNrrrrctjj|r|std|z|j }| t d||j ||j|j|tjj|rtj|tj||j||S)a)Converts the model to the CTranslate2 format. Arguments: output_dir: Output directory where the CTranslate2 model is saved. vmap: Optional path to a vocabulary mapping file that will be included in the converted model directory. quantization: Weight quantization scheme (possible values are: int8, int8_float32, int8_float16, int8_bfloat16, int16, float16, bfloat16, float32). force: Override the output directory if it already exists. Returns: Path to the output directory. Raises: RuntimeError: If the output directory already exists and :obj:`force` is not set. NotImplementedError: If the converter cannot convert this model to the CTranslate2 format. z;output directory %s already exists, use --force to overridez>* %eM  ZZ\  %N     2 24 86 77>>* % MM* % J #rct)N)r%)rs rr$zConverter._loadks !##r)NNF)__name__ __module__ __qualname____doc__ staticmethodargparseArgumentParserr Namespacestrrrboolrabcabstractmethodr$rrrr s*("9"9h>U>U2 h&8&8 S (#&* 00sm0sm 0  0 0d $$rr) r9r4r r)typingrctranslate2.specs.model_specrrABCrr;rrr?s)  Hb$b$r