+
    Pj                     \   R t ^ RIHt ^ RIt^ RIHtHt ]! ]4      P                  P                  R,          t	 ! R R]4      t
 ! R R]
4      t ! R	 R
]
4      t ! R R]
4      t ! R R]
4      t ! R R]
4      t ! R R]
4      t ! R R]
4      t ! R R]
4      t ! R R]
4      tR R ltR# )z3Configuration models for loading YAML config files.)PathN)	BaseModel
ConfigDictconfigc                   (    ] tR t^t]! RR7      tRtR# )StrictModelforbid)extra N)__name__
__module____qualname____firstlineno__r   model_config__static_attributes__r
       p/Users/ahmed/devFolder/Ultron/claude-voice/gateway/.venv/lib/python3.14/site-packages/pocket_tts/utils/config.pyr   r      s    H-Lr   r   c                   ,   a  ] tR t^t o V 3R ltRtV tR# )
FlowConfigc                2   < V ^8  d   Qh/ S[ ;R&   S[ ;R&   # )   dimdepthint)format__classdict__s   "r   __annotate__FlowConfig.__annotate__   s     	H J r   r
   Nr   r   r   r   __annotate_func__r   __classdictcell__r   s   @r   r   r            r   r   c                   ,   a  ] tR t^t o V 3R ltRtV tR# )FlowLMTransformerConfigc                V   < V ^8  d   Qh/ S[ ;R&   S[ ;R&   S[ ;R&   S[ ;R&   S[ ;R&   # )r   hidden_scale
max_periodd_model	num_heads
num_layersr   )r   r   s   "r   r   $FlowLMTransformerConfig.__annotate__   s>      O  L  N	 
 O r   r
   Nr   r"   s   @r   r%   r%      r#   r   r%   c                   ,   a  ] tR t^t o V 3R ltRtV tR# )LookupTablec                J   < V ^8  d   Qh/ S[ ;R&   S[ ;R&   S[;R&   S[;R&   # )r   r   n_bins	tokenizertokenizer_path)r   str)r   r   s   "r   r   LookupTable.__annotate__   s2     	H K  N  	 r   r
   Nr   r"   s   @r   r.   r.      r#   r   r.   c                   8   a  ] tR t^&t o RtRtRtV 3R ltRtV t	R# )FlowLMConfigz/Root configuration model for YAML config files.NFc                p   < V ^8  d   Qh/ S[ ;R&   S[;R&   S[;R&   S[;R&   S[ R,          ;R&   S[;R&   # )r   dtypeflowtransformerlookup_tableNweights_pathinsert_bos_before_voice)r3   r   r%   r.   bool)r   r   s   "r   r   FlowLMConfig.__annotate__&   sV      J    )(    *#  ") r   r
   )
r   r   r   r   __doc__r<   r=   r    r   r!   r"   s   @r   r6   r6   &   s     9  $L$)  r   r6   c                   ,   a  ] tR t^6t o V 3R ltRtV tR# )SEANetConfigc                   < V ^8  d   Qh/ S[ ;R&   S[ ;R&   S[ ;R&   S[ ;R&   S[S[ ,          ;R&   S[ ;R&   S[ ;R&   S[ ;R&   S[ ;R	&   S[;R
&   S[ ;R&   # )r   	dimensionchannels	n_filtersn_residual_layersratioskernel_sizeresidual_kernel_sizelast_kernel_sizedilation_basepad_modecompress)r   listr3   )r   r   s   "r   r   SEANetConfig.__annotate__6   s     N M  N  	 
 I          M  M r   r
   Nr   r"   s   @r   rB   rB   6   r#   r   rB   c                   0   a  ] tR t^Et o RtV 3R ltRtV tR# )MimiTransformerConfigg     @c                   < V ^8  d   Qh/ S[ ;R&   S[ ;R&   S[S[ R3,          ;R&   S[ ;R&   S[ ;R&   S[;R&   S[ ;R&   S[;R	&   S[ ;R
&   # )r   r)   input_dimension.output_dimensionsr*   r+   layer_scalecontextr(   dim_feedforward)r   tuplefloat)r   r   s   "r   r   "MimiTransformerConfig.__annotate__E   sz     L   S#X&  N	 
 O    L     r   r
   N)r   r   r   r   r(   r    r   r!   r"   s   @r   rR   rR   E   s       J  r   rR   c                   ,   a  ] tR t^Rt o V 3R ltRtV tR# )QuantizerConfigc                2   < V ^8  d   Qh/ S[ ;R&   S[ ;R&   # )r   rD   output_dimensionr   )r   r   s   "r   r   QuantizerConfig.__annotate__R   s     N  r   r
   Nr   r"   s   @r   r]   r]   R   r#   r   r]   c                   <   a  ] tR t^Xt o RtRtRtRtV 3R ltRt	V t
R# )
MimiConfigz4Root configuration model for Mimi YAML config files.Nc                   < V ^8  d   Qh/ S[ ;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[ R,          ;R	&   S[R,          ;R
&   S[R,          ;R&   # )r   r8   sample_raterE   
frame_rateseanetr:   	quantizerNr<   	inner_dim	outer_dim)r3   r   rZ   rB   rR   r]   )r   r   s   "r   r   MimiConfig.__annotate__X   s      J    M      '& " # $ *#% & Tz ' ( Tz ) r   r
   )r   r   r   r   r@   r<   rh   ri   r    r   r!   r"   s   @r   rb   rb   X   s      >"  $L I I)  r   rb   c                   @   a  ] tR t^ot o RtRtRtRtRtV 3R lt	Rt
V tR# )ConfigNFc                   < V ^8  d   Qh/ S[ ;R&   S[;R&   S[R,          ;R&   S[R,          ;R&   S[;R&   S[;R&   S[R,          ;R&   # )	r   flow_lmmimiNr<   "weights_path_without_voice_cloning pad_with_spaces_for_short_inputsremove_semicolons"model_recommended_frames_after_eos)r6   rb   r3   r>   r   )r   r   s   "r   r   Config.__annotate__o   sh      
  *#  ),d
9	 
 '+2  #  ),d
9 r   r
   )r   r   r   r   r<   rp   rq   rr   rs   r    r   r!   r"   s   @r   rl   rl   o   s+       $L59&-2$#59&  r   rl   c                F    V ^8  d   QhR\         \        ,          R\        /# )r   	yaml_pathreturn)r3   r   rl   )r   s   "r   r   r   y   s     ! !3: !& !r   c           	         \        V 4      p V P                  4       '       gh   V P                  \        4      '       d>   \	        R T  R\        P                  R4       Uu. uF  qP                  NK  	  up 24      h\	        R V  R24      h\        V R4      ;_uu_ 4       p\        P                  ! V4      pRRR4       \        R/ XB # u upi   + '       g   i     L ; i)zConfig file not found: z,. Did you make a typo? Available languages: z*.yamlz. Did you make a typo?rNr
   )r   existsis_relative_toCONFIGS_DIRFileNotFoundErrorglobstemopenyaml	safe_loadrl   )rv   pfconfig_dicts   &   r   load_configr   y   s    YI##K00#)) 5=NYN^N^_gNh=iNhffNh=i<jl   "9)DZ [\\	i		nnQ' 
  K   >j 
	s   CCC	)r@   pathlibr   r   pydanticr   r   __file__parentr|   r   r   r%   r.   r6   rB   rR   r]   rb   rl   r   r
   r   r   <module>r      s    9   *8n##**X5.) .
 k + *; * ; 	K 	k ! !.:[ :!r   