jRddlZddlZddlZddlmZmZmZddlZddlm Z idddddd d d d d dddddddddddddddddddd d!d"d#d$d%d%d&Z d'ee fd(Z d)Z d*Z dN)ListOptionalUnion)tqdmztiny.enzSystran/faster-whisper-tiny.entinyzSystran/faster-whisper-tinyzbase.enzSystran/faster-whisper-base.enbasezSystran/faster-whisper-basezsmall.enzSystran/faster-whisper-small.ensmallzSystran/faster-whisper-smallz medium.enz Systran/faster-whisper-medium.enmediumzSystran/faster-whisper-mediumzlarge-v1zSystran/faster-whisper-large-v1zlarge-v2zSystran/faster-whisper-large-v2zlarge-v3zSystran/faster-whisper-large-v3largezdistil-large-v2z&Systran/faster-distil-whisper-large-v2zdistil-medium.enz'Systran/faster-distil-whisper-medium.enzdistil-small.enz&Systran/faster-distil-whisper-small.enzdistil-large-v3z&Systran/faster-distil-whisper-large-v3zdistil-large-v3.5z$distil-whisper/distil-large-v3.5-ct2z,mobiuslabsgmbh/faster-whisper-large-v3-turbo)zlarge-v3-turboturboreturncNttS)z&Returns the names of available models.)list_MODELSkeysw/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib/python3.11/site-packages/faster_whisper/utils.pyavailable_modelsr"s    rctjtjtjt dS)z)Returns the path to the assets directory.assets)ospathjoindirnameabspath__file__rrrget_assets_pathr's4 7<<(A(ABBH M MMrc*tjdS)zReturns the module logger.faster_whisper)logging getLoggerrrr get_loggerr#,s  - . ..rF size_or_id output_dirlocal_files_only cache_dirrevisionuse_auth_tokenc `tjd|r|}n[t|}|?t d|ddtgd}||t|d}| ||d<d |d <|||d <|||d <tj |fi|S) a|Downloads a CTranslate2 Whisper model from the Hugging Face Hub. Args: size_or_id: Size of the model to download from https://huggingface.co/Systran (tiny, tiny.en, base, base.en, small, small.en, distil-small.en, medium, medium.en, distil-medium.en, large-v1, large-v2, large-v3, large, distil-large-v2, distil-large-v3), or a CTranslate2-converted model ID from the Hugging Face Hub (e.g. Systran/faster-whisper-large-v3). output_dir: Directory where the model should be saved. If not set, the model is saved in the cache directory. local_files_only: If True, avoid downloading the file and return the path to the local cached file if it exists. cache_dir: Path to the folder where cached files are stored. revision: An optional Git revision id which can be a branch name, a tag, or a commit hash. use_auth_token: HuggingFace authentication token or True to use the token stored by the HuggingFace config folder. Returns: The path to the downloaded model. Raises: ValueError: if the model size is invalid. z.*/.*NzInvalid model size 'z', expected one of: z, )z config.jsonzpreprocessor_config.jsonz model.binztokenizer.jsonz vocabulary.*)r&allow_patterns tqdm_classr( local_dirFlocal_dir_use_symlinksr'token) rematchrget ValueErrorrr disabled_tqdmhuggingface_hubsnapshot_download) r$r%r&r'r(r)repo_idr+kwargss rdownload_modelr91s@ x*%%++j)) ?*::tyy888:  N-(# F({+0'('{!(w  ,W ? ? ? ??r.secondsalways_include_hoursdecimal_markerc|dks Jdt|dz}|dz}||dzz}|dz}||dzz}|dz}||dzz}|s|dkr|ddnd }||dd|d||d S) Nrznon-negative timestamp expectedg@@i6i`i02d:03d)round)r;r<r= millisecondshoursminutes hours_markers rformat_timestamprHws a<<<:<<<6)**L I %EEI%%Lf$GGf$$Le#GGeO#L&:Oeaiie?????RL VVVVwVVNVLVVVrceZdZfdZxZS)r4cDd|d<tj|i|dS)NTdisable)super__init__)selfargsr8 __class__s rrMzdisabled_tqdm.__init__s/ y$)&)))))r)__name__ __module__ __qualname__rM __classcell__)rPs@rr4r4s8*********rr4segmentscptdt|D|r|ddndS)Nc3VK|]$}t|dD] }|dV %dS)wordsendN)reversed).0sws r zget_end..s?LLax' 7K7KLL!5LLLLLLLrrY)nextrZ)rUs rget_endrasC LL8H--LLL'1 UT  r)NFNNN)Fr:)r!rr0typingrrrr5 tqdm.autorrstrrrr#boolr9floatrHr4dictrarrrrhs ((((((((((  /  ) /  )   1    +  3  - 1 1 1  . ? A ?  ?! "?# $E ;'   . $s)    NNN ///!%"#"15 C@C@C@ C@C@} C@ sm C@ U39-. C@C@C@C@P"'  .*****D*** d4jXe_r