+
    TV-jü  ã                   óf   € ^ RI Ht ^ RIHt ^ RIHt ^ RIHt RRRRRR/tRR	 R
 llt	RR R llt
R# )é    )ÚPath)ÚUnionN)Úbase_load_modelÚsileroÚ
silero_vadz
silero-vadÚfsmnc                ó~   € V ^8„  d   QhR\         \        \        3,          R\        R\        R\        P
                  /# ©é   Ú
model_pathÚlazyÚstrictÚreturn©r   Ústrr   ÚboolÚnnÚModule)Úformats   "Úd/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_audio/vad/utils.pyÚ__annotate__r      s9   € ÷ ñ Ü”cœ4�iÕ ðÜ(,ðÜ>Bðä‡Y�Yñó    c                ó4   € \        RRV RRR\        RVRV/VB # )a‰  
Load and initialize a VAD/diarization 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.
r   ÚcategoryÚvadÚmodel_remappingr   r   © )r   ÚMODEL_REMAPPING©r   r   r   Úkwargss   &&&,r   Ú
load_modelr!      sI   € ô ñ Øðàðô (ðð ð	ð
 ðð ñð r   c                ó~   € V ^8„  d   QhR\         \        \        3,          R\        R\        R\        P
                  /# r
   r   )r   s   "r   r   r   $   s?   € ÷ Fñ FÜ”cœ4�iÕ ðFÜ(,ðFÜ>BðFä‡Y�YñFr   c                ó"   € \        V 3RVRV/VB # )a=  
Load a VAD/diarization model from a local path or HuggingFace repository.

This is the main entry point for loading VAD 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:
        - revision (str): HuggingFace revision/branch to use
        - force_download (bool): Force re-download of model files

Returns:
    nn.Module: The loaded and initialized model.

Example:
    >>> from mlx_audio.vad import load
    >>> model = load("mlx-community/diar_streaming_sortformer_4spk-v2.1-fp16")
    >>> result = model.generate("audio.wav", verbose=True)
r   r   )r!   r   s   &&&,r   Úloadr$   $   s   € ô2 �jÑE tÐE°FÐE¸fÑEÐEr   )FF)Úpathlibr   Útypingr   Úmlx.nnr   Úmlx_audio.utilsr   r   r!   r$   r   r   r   Ú<module>r)      s4   ðÝ Ý å å +à˜\¨<¸ÀvÈvÐV€÷÷2Fñ Fr   