NjddlZddlmZmZmZddlZddlmZejeZ edZ GddeZ Gddejee Z y) N)Generic NamedTupleTypeVar)nnPreparedc,eZdZUejed<y) TokenizedTexttokensN)__name__ __module__ __qualname__torchTensor__annotations__m/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/pocket_tts/conditioners/base.pyr r s LLrr c \eZdZdZ d dedededeffd Zdedejfd Z xZ S) BaseConditioneraBase 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. dim output_dim output_bias force_linearcXt|||_||_|s||k7sJ|rJyN)super__init__rr)selfrrrr __class__s rrzBaseConditioner.__init__s8 $sj000;rinputsreturnc$|j|Sr)_get_condition)rr s rforwardzBaseConditioner.forward%s""6**r)FT) r r r __doc__intboolrr rrr$ __classcell__)rs@rrrsKZ^$'6:RV+m+ +rr)loggingtypingrrrrr getLoggerr loggerrr Modulerrrrr.sU//    8 $ : J+bii!2+r