o
    ¡¿¯iX  ã                   @   sN   d Z ddlmZmZ ddlmZmZ ddlZddlm	Z	 G dd„ de	eƒZ
dS )zGAN-based TTS abstrast class.é    )ÚABCÚabstractmethod)ÚDictÚUnionN)ÚAbsTTSc                	   @   s@   e Zd ZdZedeeeej	eeej	f e
f f fdd„ƒZdS )Ú	AbsGANTTSz#GAN-based TTS model abstract class.Úreturnc                 O   s   t ‚)z'Return generator or discriminator loss.)ÚNotImplementedError)ÚselfÚforward_generatorÚargsÚkwargs© r   úO/home/ubuntu/.local/lib/python3.10/site-packages/espnet2/gan_tts/abs_gan_tts.pyÚforward   s   zAbsGANTTS.forwardN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   Ústrr   ÚtorchÚTensorÚintr   r   r   r   r   r      s     ûr   )r   Úabcr   r   Útypingr   r   r   Úespnet2.tts.abs_ttsr   r   r   r   r   r   Ú<module>   s   