+
    TV-jã	  ã                   ó”   € ^ RI t ^ RIHt ^ RIHt ^ RIt] ! R R4      4       tR t	R t
] ! R R4      4       t] ! R	 R
4      4       tR# )é    N)Ú	dataclassc                   ó0   a € ] tR t^t o ]R 4       tRtV tR# )ÚBaseModelArgsc                ó°   € T ! R/ VP                  4        UUu/ uF.  w  r#V\        P                  ! V 4      P                  9   g   K,  W#bK0  	  uppB # u uppi )N© )ÚitemsÚinspectÚ	signatureÚ
parameters)ÚclsÚparamsÚkÚvs   &&  Új/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_audio/tts/models/base.pyÚ	from_dictÚBaseModelArgs.from_dict
   s[   € áñ 
ð #ŸL™LœNôá*‘D�AØœ×)Ò)¨#Ó.×9Ñ9Ñ9ô �’Ù*òñ
ð 	
ùós   ˜)AÁAr   N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Úclassmethodr   Ú__static_attributes__Ú__classdictcell__©Ú__classdict__s   @r   r   r      s   ø‡ € àñ
ó ö
ó    r   c                 ót   € V P                   p\        V4      ^8w  d   R# Vw  r#pW#8¼  d   W$8¼  d	   W48X  d   R# R# )é   FT)ÚshapeÚlen)Úarrr   Úout_channelsÚkHÚKWs   &    r   Úcheck_array_shaper%      s<   € Ø�I‰I€Eô ˆ5ƒz�Q„Ùà Ñ€L�bð 	Ô Ô!3¸"¼(Ùár   c                ó|  € \        V \        P                  4      '       g   \        P                  ! V 4      p V P                  ^ ,          p\	        W!,          4      p\        P
                  ! V4      p\        P                  ! ^ V^,
          V4      p\        P                  ! V4      P                  \        P                  4      p\        P                  ! V^,           V^,
          4      pWV,
          pRV,
          p	V	P                  R^4      W,          ,          VP                  R^4      W,          ,          ,           p
V
# )z_
Adjust the speed of the audio by resampling
speed_factor > 1: faster
speed_factor < 1: slower
g      ð?éÿÿÿÿ)Ú
isinstanceÚmxÚarrayr   ÚintÚarangeÚlinspaceÚfloorÚastypeÚint32ÚminimumÚreshape)Úaudio_arrayÚspeed_factorÚ
old_lengthÚ
new_lengthÚold_indicesÚnew_indicesÚindices_floorÚindices_ceilÚweights_ceilÚweights_floorÚresults   &&         r   Úadjust_speedr>   %   sö   € ô �k¤2§8¡8×,Ò,Ü—h’h˜{Ó+ˆð ×"Ñ" 1Õ%€JÜ�ZÕ.Ó/€Jô —)’)˜JÓ'€KÜ—+’+˜a ¨a¥°Ó<€Kô —H’H˜[Ó)×0Ñ0´·±Ó:€MÜ—:’:˜m¨aÕ/°¸aµÓ@€LØÕ.€LØ˜,Õ&€Mð 	×Ñ˜b !Ó$ {Õ'AÕAØ
×
Ñ
˜r 1Ó
%¨Õ(AÕ
Aõ	Bð ð
 €Mr   c                   ó4   a € ] tR t^Gt o RtRtV 3R ltRtV tR# )ÚGenerationResultFc                óÊ   <€ V ^8„  d   Qh/ S[ P                  ;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R	&   S[;R
&   S[;R&   S[;R&   S[;R&   # )é   ÚaudioÚsamplesÚsample_rateÚsegment_idxÚtoken_countÚaudio_durationÚreal_time_factorÚpromptÚaudio_samplesÚprocessing_time_secondsÚpeak_memory_usageÚis_streaming_chunkÚis_final_chunk)r)   r*   r+   ÚstrÚfloatÚdictÚbool)Úformatr   s   "€r   Ú__annotate__ÚGenerationResult.__annotate__G   s­   ø‡ ‚ á�8‰8�Oñ ñ �Lñ ñ Ññ	 ñ
 Ññ ñ Ññ ñ Ññ ñ Ññ ñ �Lñ ñ Ññ ñ #Ñ"ñ ñ Ññ ñ Ñ$ñ ñ Ñ ò r   r   N©	r   r   r   r   rN   rO   Ú__annotate_func__r   r   r   s   @r   r@   r@   G   s   ø‡ € ð  %ÐØ €N÷ ƒ r   r@   c                   ó4   a € ] tR t^Xt o RtRtV 3R ltRtV tR# )ÚBatchGenerationResultFc                ó¦   <€ V ^8„  d   Qh/ S[ P                  ;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R	&   S[;R
&   # )rB   rC   Úsequence_idxrD   rE   rG   rH   rL   rM   rN   rO   )r)   r*   r+   rP   rQ   rS   )rT   r   s   "€r   rU   Ú"BatchGenerationResult.__annotate__X   s‡   ø‡ ‚ á�8‰8�Oñ ñ Ññ ñ �Lñ	 ñ
 Ññ ñ Ññ ñ Ññ ñ #Ñ"ñ ñ Ññ ñ Ñ$ñ ñ Ñ ò r   r   NrW   r   s   @r   rZ   rZ   X   s   ø‡ € ð  %ÐØ €N÷ ƒ r   rZ   )r	   Údataclassesr   Úmlx.coreÚcorer)   ÚnumpyÚnpr   r%   r>   r@   rZ   r   r   r   Ú<module>rc      sg   ðÛ Ý !å Û ð ÷	
ð 	
ó ð	
òò ðD ÷!ð !ó ð!ð  ÷
!ð 
!ó ò
!r   