o di@sjddlZddlZddlZddlmZmZmZddlZddlm Z idddddd d d d d dddddddddddddddddddd d!d"d#d$d%d%d&Z d'ee fd(d)Z d*d+Z d,d-Z  .   dBd/e d0ee d1ed2ee d3ee d4eee eff d5d6Z . 7dCd8ed9ed:e d'e fd;d<ZGd=d>d>e Zd?eed'eefd@dAZdS)DN)ListOptionalUnion)tqdmztiny.enzSystran/faster-whisper-tiny.entinyzSystran/faster-whisper-tinyzbase.enzSystran/faster-whisper-base.enbasezSystran/faster-whisper-basezsmall.enzSystran/faster-whisper-small.enZsmallzSystran/faster-whisper-smallz medium.enz Systran/faster-whisper-medium.enZmediumzSystran/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-turboZturboreturncCs ttS)z&Returns the names of available models.)list_MODELSkeysr r 4/home/jaya/work/projects/WHISPER/faster_asr/utils.pyavailable_models"s rcCstjtjtjtdS)z)Returns the path to the assets directory.Zassets)ospathjoindirnameabspath__file__r r r rget_assets_path'srcCs tdS)zReturns the module logger. faster_asr)logging getLoggerr r r r get_logger,s rF size_or_id output_dirlocal_files_only cache_dirrevisionuse_auth_tokenc Cstd|r |}nt|}|durtd|dtfgd}||t|d}|dur2||d<|dur:||d<|durB||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.*/.*Nz,Invalid model size '%s', expected one of: %sz, )z config.jsonzpreprocessor_config.jsonz model.binztokenizer.jsonz vocabulary.*)rallow_patternsZ tqdm_classrZ local_dirrtoken) rematchr get ValueErrorrr disabled_tqdmhuggingface_hubsnapshot_download) rrrrrr repo_idr!kwargsr r rdownload_model1s,   r,.secondsalways_include_hoursdecimal_markercCs|dksJdt|d}|d}||d8}|d}||d8}|d}||d8}|s2|dkr8|ddnd }||dd|d||d S) Nrznon-negative timestamp expectedg@@i6i`i02d:Z03d)round)r.r/r0Z millisecondshoursminutesZ hours_markerr r rformat_timestampvs    r7cseZdZfddZZS)r'csd|d<tj|i|dS)NTdisable)super__init__)selfargsr+ __class__r rr:szdisabled_tqdm.__init__)__name__ __module__ __qualname__r: __classcell__r r r=rr'sr'segmentscCs*tddt|D|r|ddSdS)Ncss*|]}t|dD]}|dVq qdS)wordsendN)reversed).0swr r r s(zget_end..rE)nextrF)rCr r rget_ends rM)NFNNN)Fr-)rrr#typingrrrr(Z tqdm.autorr strrrrboolr,floatr7r'dictrMr r r rs       G