+
    &j                     |    ^ RI t ^ RIHt ^ RIHt ^ RIHt ^ RIH	t	 ^ RI
Ht ^RIHt ]P                  3R R lltR# )    N)Path)snapshot_download)tree_unflatten)whisperc                d    V ^8  d   QhR\         R\        P                  R\        P                  /# )   path_or_hf_repodtypereturn)strmxDtyper   Whisper)formats   "o/Users/jameslopez/projects/CWCArchive/cwc-podcast/.venv/lib/python3.14/site-packages/mlx_whisper/load_models.py__annotate__r      s.         88  __     c                 d  a
 \        V 4      pVP                  4       '       g   \        \        V R 7      4      p\        \	        VR,          4      R4      ;_uu_ 4       p\
        P                  ! VP                  4       4      pVP                  RR4       VP                  RR4      pRRR4       \        P                  ! R
/ XB pVR,          pVP                  4       '       g
   VR,          p\        P                  ! \	        V4      4      o
\        P                  ! Wa4      pXe"   V
3R lp	\        P                  ! V3/ VBR	V	/B  \!        \#        S
P%                  4       4      4      o
VP'                  S
4       \        P(                  ! VP+                  4       4       V#   + '       g   i     EL; i))repo_idzconfig.jsonr
model_typeNquantizationzweights.safetensorszweights.npzc                 x   < \        V\        P                  \        P                  34      ;'       d	    V  R 2S9   # )z.scales)
isinstancennLinear	Embedding)pmweightss   &&r   <lambda>load_model.<locals>.<lambda>&   s6    A		2<<'@A ) )#W()r   class_predicate )r   existsr   openr   jsonloadsreadpopr   ModelDimensionsr   loadr   r   quantizer   listitemsupdateeval
parameters)r	   r
   
model_pathfconfigr   
model_argswfmodelr#   r    s   &&        @r   
load_modelr9      sB    o&J+ODE
	c*},-s	3	3qAFFH%

<&zz.$7 
4
 ((262J	+	+B99;;-'ggc"gGOOJ.E) 	 	EK\K?KT'--/23G	LLGGEL1 
4	3	3s   A	FF/	)r'   pathlibr   mlx.corecorer   mlx.nnr   huggingface_hubr   	mlx.utilsr    r   float32r9   r$   r   r   <module>rB      s/        - $ 
 jj   r   