
    h                         S SK r S SKrS SKrS SKJrJr  S rS rS r	S r
S\R                  S\S	\R                  4S
 jrS rS rg)    N)download_url_to_filemodule_utilsc                      g)N)&_-!'(),.:;? abcdefghijklmnopqrstuvwxyz r,       W/var/www/auris/envauris/lib/python3.13/site-packages/torchaudio/pipelines/_tts/utils.py
_get_charsr/      s    'r-   c                      g)N)`r   r   r   r	   r
   r   r   r   r   r   r   r   AAAA0AA1AA2AEAE0AE1AE2AHAH0AH1AH2AOAO0AO1AO2AWAW0AW1AW2AYAY0AY1AY2BCHDDHEHEH0EH1EH2ERER0ER1ER2EYEY0EY1EY2FGHHIHIH0IH1IH2IYIY0IY1IY2JHKLMNNGOWOW0OW1OW2OYOY0OY1OY2PRSSHTTHUHUH0UH1UH2UWUW0UW1UW2VWYZZHr,   r,   r-   r.   _get_phonesr   3   s    ar-   c                 F   [         R                  " U  Vs/ s H  n[        U5      PM     sn[         R                  S9nU  Vs/ s H  n[         R                  " U5      PM     nn[         R                  R
                  R                  R                  USS9nX24$ s  snf s  snf )N)dtypeT)batch_first)torchtensorlenint32nnutilsrnnpad_sequence)indicesr   lengthsvaluess       r.   
_to_tensorr      sx    llG4GqCFG4EKKHG'./w!ell1owF/XX^^,,V,FF? 5/s   B Bc                 X   [         R                  " S5      (       d  [        S5      eSSKJn  SSKJnJnJn  [        R                  " S5      nUR                  nUR                  [        R                  5         SU  3n[        R                  R!                  ["        R$                  R'                  5       S5      n	[        R(                  " U	SS	9  [        R                  R!                  X5      n
[        R                  R+                  U
5      (       d  Uc  0 OUn[-        X40 UD6  ["        R.                  R1                  X4U/5         UR3                  U
5      sS S S 5        UR                  U5        $ ! , (       d  f       O= f UR                  U5        g ! UR                  U5        f = f)
Ndpz3DeepPhonemizer is not installed. Please install it.r   )
Phonemizer)PreprocessorLanguageTokenizerSequenceTokenizerzKhttps://public-asai-dl-models.s3.eu-central-1.amazonaws.com/DeepPhonemizer/checkpointsT)exist_ok)
_mod_utilsis_module_availableRuntimeErrordp.phonemizerr   dp.preprocessing.textr   r   r   logging	getLoggerlevelsetLevelINFOospathjoinr   hubget_dirmakedirsexistsr   serializationsafe_globalsfrom_checkpoint)file	dl_kwargsr   r   r   r   logger
orig_levelurl	directoryr   s              r.   _load_phonemizerr      s:   ))$//PQQ(XX t$FJ
OOGLL!$[\`[abGGLL!2!2!4mD	
I-ww||I,ww~~d##'/YI 8i8  --|Pa.bc--d3 dc 	
# dcc 	
#
#s%   9CF E2	F 2
F <F F)waveformbitsreturnc                     [         R                  " U SS5      n U S-   SU-  S-
  -  S-  n [         R                  " U SSU-  S-
  5      R                  5       $ )z6Transform waveform [-1, 1] to label [0, 2 ** bits - 1]   g      ?   r   )r   clampint)r   r   s     r.   _unnormalize_waveformr      sS    {{8R+H31d7Q;/!3H;;xAtGaK04466r-   c                 |    0 SS_SS_SS_SS_S	S_S
S_SS_SS_SS_SS_SS_SS_SS_SS_SS_SS_SS_SSSS U S!.E$ )"Nmask_paddingFn_melsP   n_frames_per_stepr   symbol_embedding_dim   encoder_embedding_dimencoder_n_convolution   encoder_kernel_size   decoder_rnn_dimi   decoder_max_stepi  decoder_dropoutg?decoder_early_stoppingTattention_rnn_dimattention_hidden_dim   attention_location_n_filter    attention_location_kernel_size   attention_dropout
prenet_dim   g      ?)postnet_n_convolutionpostnet_kernel_sizepostnet_embedding_dimgate_thresholdn_symbolr,   )	n_symbolss    r.   _get_taco_paramsr      s    " 	Q 		
 	  	  	q 	4 	D 	3 	!$ 	T 	 	&r 	)"  	S!" 	c#$ "# !$- r-   c                       / SQSSSSSSSSSS	.
$ )
N)r   r      r   i  
   r   r   r   r   )
upsample_scales	n_classes
hop_lengthn_res_blockn_rnnn_fckernel_sizen_freqn_hiddenn_outputr,   r,   r-   r.   _get_wrnn_paramsr      s*    % r-   )r   r   r   torchaudio._internalr   r   r   r/   r   r   r   Tensorr   r   r   r   r,   r-   r.   <module>r      sT     	  Q(VbJ$27ELL 7 7 76r-   