o ´‹ jáã@s,ddlmZddlmZGdd„deƒZdS)é)ÚCoqpit)ÚBaseTrainerModelcs2eZdZdZdZ‡fdd„Zdefdd„Z‡ZS)Ú BaseVocoderalBase `vocoder` class. Every new `vocoder` model must inherit this. It defines `vocoder` specific functions on top of `Model`. Notes on input/output tensor shapes: Any input or output tensor of the model must be shaped as - 3D tensors `batch x time x channels` - 2D tensors `batch x channels` - 1D tensors `batch x 1` Úvocodercstƒ ¡| |¡dS)N)ÚsuperÚ__init__Ú_set_model_args)ÚselfÚconfig©Ú __class__©úR/home/kuhnn/.local/lib/python3.10/site-packages/TTS/vocoder/models/base_vocoder.pyrs zBaseVocoder.__init__r cCs¬d|jjvrRd|vr9| |¡\}|_}||j_t|jdƒr5||j_d|vr*|jj|_d|vr7|jj|_dSdSdS||_d|vrE|jj|_d|vrP|jj|_dSdSt dƒ‚)aSetup model args based on the config type. If the config is for training with a name like "*Config", then the model args are embeded in the config.model_args If the config is for the model with a name like "*Args", then we assign the directly. ÚConfigÚ charactersÚ model_argsÚ model_paramsz(config must be either a *Config or *ArgsN) r Ú__name__Úget_charactersr Ú num_charsÚhasattrrÚargsrÚ ValueError)r r Ú_rr r rrs&   ú ÿzBaseVocoder._set_model_args) rÚ __module__Ú __qualname__Ú__doc__Ú MODEL_TYPErrrÚ __classcell__r r r rrs  rN)ÚcoqpitrÚ TTS.modelrrr r r rÚs