+ TV-j (^RIHt^RIHtHtHt^RIHt^RI H t ^RI t ^RI HtHtHtRt/RRbRRbRR bR R bR R bR R bR RbRRbRRbRRbRRbRRbRRbRRbRRbRRbtRRlt]],]R]P*3RRlltR!RRlltR!RR lltR#)")Path)AnyOptionalUnionN)base_load_modelget_model_path load_configi> cohere_asr fireredasr2glmglmasr sensevoicevoxtralvoxtral_realtime vibevoice vibevoice_asr qwen3_asr fun_asr_nanocanary moonshinemmsgranite_speechgranite_speech_nar qwen2_audiomega_asrcpV^8dQhR\PR\R\R\P/#)audioorig_sr target_srreturn)npndarrayint)formats"d/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_audio/stt/utils.py __annotate__r' s0>>"**>s>s>rzz>c$^RIHpV!WV^R7#)r)resample_audioaxis)mlx_audio.utilsr*)rrr _resample_audios&&& r&r*r* sA 591 ==r(FcPV^8dQhR\R\R\P/#)rfilesrdtype)strr$mxDtype)r%s"r&r'r'&s*55 5 5 88 5r(c^RIHpV!VRR7wrVWa8wd \WVV4p\P!WSR7P ^R7#)a Open an audio file and read as mono waveform, resampling as necessary Parameters ---------- file: str The audio file to open sr: int The sample rate to resample the audio if necessary Returns ------- A NumPy array containing the audio waveform, in float32 dtype. )readT) always_2d)r2r+)mlx_audio.audio_ior7r*r4arraymean)r0r1 from_stdinr2 audio_readr sample_rates&&&& r& load_audior?&sE*6#DD9Eu26 88E ' , ,! , 44r(c V^8dQhR\\\3,R\R\R\R\ P /#r model_pathlazystrictkwargsr!rr3rboolrnnModule)r%s"r&r'r'CsGc4i     YY r(c 4\RRVRRR\RVRV/VB#)a~ Load and initialize an STT model from a given path. Args: model_path: The path or HuggingFace repo to load the model from. lazy: If False, evaluate model parameters immediately. strict: If True, raise an error if any weights are missing. **kwargs: Additional keyword arguments (revision, force_download). Returns: nn.Module: The loaded and initialized model. rBcategorysttmodel_remappingrCrD)rMODEL_REMAPPINGrBrCrDrEs&&&,r& load_modelrQCsI$    (         r(c V^8dQhR\\\3,R\R\R\R\ P /#rArF)r%s"r&r'r'_sNFFc4i F F F F YY Fr(c "\V3RVRV/VB#)a3 Load a speech-to-text model from a local path or HuggingFace repository. This is the main entry point for loading STT models. It automatically detects the model type and initializes the appropriate model class. Args: model_path: The local path or HuggingFace repo ID to load from. lazy: If False, evaluate model parameters immediately. strict: If True, raise an error if any weights are missing. **kwargs: Additional keyword arguments such as `revision` and `force_download`. Returns: nn.Module: The loaded and initialized model. rCrD)rQrPs&&&,r&loadrT_s. j Et EF Ef EEr()FF)pathlibrtypingrrrmlx.corecorer4mlx.nnrHnumpyr"r-rrr SAMPLE_RATErOr*r3float32r?rQrTrNr(r&r]s''HH ,= 8,  y  * N h 5&.=  !(> jj 5:8FFr(