+
    TV-jà  ã                   ój   € ^ RI Ht ^ RIHtHt ^ RIHt ^ RIHt Rt	RRRR/t
RR R	 lltRR
 R lltR# )é    )ÚPath)ÚOptionalÚUnionN)Úbase_load_modeli€>  z
ecapa-tdnnÚ
ecapa_tdnnc                ó~   € 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/lid/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 LID 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ÚlidÚ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 language identification model from a local path or HuggingFace repository.

This is the main entry point for loading LID 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.lid import load
    >>> model = load("facebook/mms-lid-256")
    >>> results = model.predict(audio)
r   r   )r    r   s   &&&,r   Úloadr#   )   s   € ô2 �jÑE tÐE°FÐE¸fÑEÐEr   )FF)Úpathlibr   Útypingr   r   Úmlx.nnr   Úmlx_audio.utilsr   ÚSAMPLE_RATEr   r    r#   r   r   r   Ú<module>r)      s:   ðÝ ß "å å +à€ð �,Ø�,ð€÷÷2Fñ Fr   