ritddlZddlZddlZddlmZmZmZddlZddlm Z idddddd d d d d dddddddddddddddddddd d!d"d#d$d%d%d&Z d'ee fd(Z d)Z d*Z d:d+e d,ee d-ed.ee d/ee d0eee eff d1Z d;d2ed3ed4e d'e fd5ZGd6d7e Zd8eed'eefd9Zy)<N)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-turboturboreturnc<ttjS)z&Returns the names of available models.)list_MODELSkeysJ/home/jaya/work/projects/FASTER-ASR-WHISPER/faster-asr/faster_asr/utils.pyavailable_modelsr"s   rctjjtjjtjj t dS)z)Returns the path to the assets directory.assets)ospathjoindirnameabspath__file__rrrget_assets_pathr's1 77<<(ABH MMrc,tjdS)zReturns the module logger. faster_asr)logging getLoggerrrr get_loggerr#,s   \ **r size_or_id output_dirlocal_files_only cache_dirrevisionuse_auth_tokenc :tjd|r|}nItj|}|2t d|ddj tj gd}||t|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.*/.*zInvalid model size 'z', expected one of: z, )z config.jsonzpreprocessor_config.jsonz model.binztokenizer.jsonz vocabulary.*)r&allow_patterns tqdm_classr( local_dirr'token) rematchrget ValueErrorrr disabled_tqdmhuggingface_hubsnapshot_download) r$r%r&r'r(r)repo_idr+kwargss rdownload_modelr81s@ xx*%++j) ?tyy8:  N-(# F({'{!(w  , ,W ? ??rsecondsalways_include_hoursdecimal_markerc|dk\sJdt|dz}|dz}||dzz}|dz}||dzz}|dz}||dzz}|s|dkDr|ddnd }||dd|d||d S) Nrznon-negative timestamp expectedg@@i6i`i02d:03d)round)r9r:r; millisecondshoursminutes hours_markers rformat_timestamprFvs a<:::<6)*L I %EEI%%Lf$GGf$$Le#GGeO#L&:eaieC[?RL . QwsmN3CLQTCUVrceZdZfdZxZS)r3c.d|d<t||i|y)NTdisable)super__init__)selfargsr7 __class__s rrKzdisabled_tqdm.__init__s y $)&)r)__name__ __module__ __qualname__rK __classcell__)rNs@rr3r3s **rr3segmentscVtdt|D|r |ddSdS)Nc3LK|]}t|dD] }|d yw)wordsendN)reversed).0sws r zget_end..s)Lax' 7KL!5LLs"$rW)nextrX)rSs rget_endr_s7 L8H-L' U -1 r)NFNNN)F.)r!rr/typingrrrr4 tqdm.autorrstrrrr#boolr8floatrFr3dictr_rrrrgs ((  /  ) /  )   1    +  3  - 1 1 1  . ? A ?  ?! "?# $E ;' . $s) N +!%"#"15 B@B@ B@B@} B@ sm B@ U39-. B@N"'  .*D* d4jXe_r