antoniomae1234/Seed-VC-characters
6
1o
2 \}�f� � @ sF d dl mZ d dlmZ d dlmZ d dlmZ G dd� dej�Z dS )� )�TupleN)�3functional)�
sequence_maskc
sJ e Zd Z ddedededed ed4ef � fd d �
Zdd
d�Z� ZS )�InterpolateRegulatorF� N� �channels�sampling_ratios� is_discrete�
codebook_size� out_channels�groupsc s� t � �� || _|p |}t�g �}t|�dkr7|D ] }t�||ddd�} t�||�}5t�� } |� | |6| g� q|�7t�||dd�� tj |� | _ t�
||�| _|| _d S )Nr � r )�super�__init__r �nn�8ModuleList�len�Conv1d� GroupNorm�Mish�extend�append�9Sequential�model� Embedding� embeddingr10 ) �selfr r r11 r r r
r �_�module�norm�act�� __class__� �&D:\seed-vc\modules\length_regulator.pyr s 12 13 � 14z InterpolateRegulator.__init__c C sd | j r| �|�}t|��d�}tj|�dd��� |�� dd�}| � |��dd��� }|}|| |fS )N�����r � �nearest)�size�mode)15r16 r r � unsqueeze�F� interpolate� transpose�17contiguous�maxr )r �x�ylens�mask�outZolensr$ r$ r% �forward" s 18 z InterpolateRegulator.forward)Fr Nr )N) �__name__�19__module__� __qualname__�intr �boolr r5 �
__classcell__r$ r$ r" r% r s& �������r )20�typingr �torch.nnr r r, �modules.commonsr �Moduler r$ r$ r$ r% �<module> s21