j hddlZddlZddlZddlZddlmZddlmZmZGddej Z dS)N)Optional)ACCEPTED_MODEL_TYPES ModelSpecc eZdZdZedejdejfdZdejde fdZ dd e d e e d e e d e de f d Z ejdZdS) Converterz Base class for model converters.parserreturnc|ddd|ddd|d dtd |d 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 /home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib64/python3.11/site-packages/ctranslate2/converters/converter.pydeclare_argumentszConverter.declare_argumentss  T0I      t2W      (,     P     argsc\||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(s5|| O#**    rNFrrrrctj|r|std|z|}|t d||||||tj|rtj |tj || ||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 overrideNz>* % % e M  ZZ\\  %N     2 24 8 8 8666 7>>* % % & M* % % % J ###rct)N)r%)rs rr$zConverter._loadks!###r)NNF)__name__ __module__ __qualname____doc__ staticmethodargparseArgumentParserr Namespacestrrrboolrabcabstractmethodr$rrrr s**("9h>U\2 h&8 S    (#&* 000sm0sm 0  0 0000d $$$$$rr) r9r4r r)typingrctranslate2.specs.model_specrrABCrr;rrr?s  HHHHHHHHb$b$b$b$b$b$b$b$b$b$r