o j@s(ddlZddlmZGdddeZdS)N)MelganGeneratorcs<eZdZ       d fdd Zed d ZZS) FullbandMelganGeneratorPr r c stj|||||||ddS)N) in_channels out_channels proj_kernel base_channelsupsample_factors res_kernelnum_res_blocks)super__init__)selfr rrrrrr __class___/home/kuhnn/.local/lib/python3.10/site-packages/TTS/vocoder/models/fullband_melgan_generator.pyrs  z FullbandMelganGenerator.__init__cCs8||jdjj}tjj||j|jfd}||S)Nr replicate) tolayersweightdevicetorchnn functionalpadinference_padding)r cond_featuresrrr inferences  z!FullbandMelganGenerator.inference)rrrrrr r )__name__ __module__ __qualname__rr no_gradr& __classcell__rrrrrsr)r #TTS.vocoder.models.melgan_generatorrrrrrrs