+
    Pj                         ^ RI t ^ RIHtHtHt ^ RIt^ RIHt ] P                  ! ]4      t	]! R4      t
 ! R R]4      t ! R R]P                  ]]
,          4      tR# )	    N)Generic
NamedTupleTypeVar)nnPreparedc                   ,   a  ] tR t^t o V 3R ltRtV tR# )TokenizedTextc                :   < V ^8  d   Qh/ S[ P                  ;R&   # )   tokens)torchTensor)format__classdict__s   "u/Users/ahmed/devFolder/Ultron/claude-voice/gateway/.venv/lib/python3.14/site-packages/pocket_tts/conditioners/base.py__annotate__TokenizedText.__annotate__   s     LL      N)__name__
__module____qualname____firstlineno____annotate_func____static_attributes____classdictcell__)r   s   @r   r	   r	      s      r   r	   c                   X   a a ] tR t^t oRtRV3R lV 3R llltV3R lR ltRtVtV ;t	# )BaseConditionera  Base model for all conditioner modules.

Args:
    dim (int): internal dim of the model.
    output_dim (int): Output dim of the conditioner.
    force_linear (bool, optional): Force linear projection even when `dim == output_dim`.
    output_bias (bool): if True, the output projection will have a bias.
    learn_padding (bool): if True, the padding value will be learnt, zero otherwise.
c                2   < V ^8  d   QhRS[ RS[ RS[RS[/# )r   dim
output_dimoutput_biasforce_linear)intbool)r   r   s   "r   r   BaseConditioner.__annotate__   s-      $'6:RVr   c                p   < \         SV `  4        Wn        W n        V'       g	   W8w  g   Q hV'       d   Q hR # N)super__init__r    r!   )selfr    r!   r"   r#   	__class__s   &&&&&r   r*   BaseConditioner.__init__   s2     	$s000;r   c                :   < V ^8  d   QhRS[ RS[P                  /# )r   inputsreturn)r	   r   r   )r   r   s   "r   r   r&   %   s     + +m + +r   c                $    V P                  V4      # r(   )_get_condition)r+   r/   s   &&r   forwardBaseConditioner.forward%   s    ""6**r   )r    r!   )FT)
r   r   r   r   __doc__r*   r3   r   r   __classcell__)r,   r   s   @@r   r   r      s#      + + +r   r   )loggingtypingr   r   r   r   r   	getLoggerr   loggerr   r	   Moduler   r   r   r   <module>r<      sS     / /  			8	$ :J +bii!2 +r   