o 8wi @sPddlZddlZddlZddlmZmZddlZddlZddlm Z idddddd d d d d dddddddddddddddddddd d!d"d#d$d%d$iZ d&ee fd'd(Z d)d*Z d+d,Z  - d?d.e d/ee d0ed1ee fd2d3Z - 4d@d5ed6ed7e d&e fd8d9ZGd:d;d;e ZdZdS)AN)ListOptional)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-v3zlarge-v3-turboz,mobiuslabsgmbh/faster-whisper-large-v3-turboturboreturncCs ttS)z&Returns the names of available models.)list_MODELSkeysrrQ/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/faster_whisper/utils.pyavailable_models"s rcCstjtjtjtdS)z)Returns the path to the assets directory.assets)ospathjoindirnameabspath__file__rrrrget_assets_path'srcCs tdS)zReturns the module logger.faster_whisper)logging getLoggerrrrr get_logger,s rF size_or_id output_dirlocal_files_only cache_dirc Cstd|r |}nt|}|durtd|dtfgd}||td}|dur5||d<d|d <|dur=||d <z tj |fi|WStj j t j jfy{}z"t}|d |||d d |d<tj |fi|WYd}~Sd}~ww)aDownloads 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. 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.*)r allow_patterns tqdm_class local_dirFlocal_dir_use_symlinksr!zOAn error occured while synchronizing the model %s from the Hugging Face Hub: %szETrying to load the model directly from the local cache, if it exists.Tr )rematchr get ValueErrorrr disabled_tqdmhuggingface_hubsnapshot_downloadutilsHfHubHTTPErrorrequests exceptionsConnectionErrorrwarning) rrr r!repo_idr"kwargs exceptionloggerrrrdownload_model1sL    r7.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:03d)round)r9r:r; millisecondshoursminutes hours_markerrrrformat_timestamp~s    rEcseZdZfddZZS)r*csd|d<tj|i|dS)NTdisable)super__init__)selfargsr4 __class__rrrHszdisabled_tqdm.__init__)__name__ __module__ __qualname__rH __classcell__rrrKrr*sr*segmentscCs*tddt|D|r|ddSdS)Ncss*|]}t|dD]}|dVq qdS)wordsendN)reversed).0swrrr s(zget_end..rS)nextrT)rQrrrget_ends r[)NFN)Fr8)rrr&typingrrr+r/ tqdm.autorr strrrrboolr7floatrEr*dictr[rrrrs       O