
    Nj                     |    d dl Z d dl mZ d dlmZmZ  G d dej
                        Z G d dej
                        Zy)    N)nn)StreamingConv1dStreamingConvTranspose1dc                   b     e Zd ZdZd
dedededz  f fdZdej                  dedz  fd	Z	 xZ
S )ConvDownsample1dzu
    Downsampling by some integer amount `stride` using convolutions
    with a kernel size of twice the stride.
    Nstride	dimensionout_dimensionc           	      ^    t         |           ||}t        ||d|z  |ddd      | _        y )N      F	replicate)kernel_sizer   groupsbiaspad_mode)super__init__r   conv)selfr   r	   r
   	__class__s       l/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/pocket_tts/modules/resample.pyr   zConvDownsample1d.__init__   s>     %M#F
 
	    xmodel_statec                 &    | j                  ||      S N)r   r   r   r   s      r   forwardzConvDownsample1d.forward   s    yyK((r   r   __name__
__module____qualname____doc__intr   torchTensordictr   __classcell__r   s   @r   r   r      sA    

s 
s 
3: 
) )D4K )r   r   c                   b     e Zd ZdZd
dedededz  f fdZdej                  dedz  fd	Z	 xZ
S )ConvTrUpsample1dzQ
    Upsample by some integer amount `stride` using transposed convolutions.
    Nr   r	   in_dimensionc                 \    t         |           ||}t        ||d|z  ||d      | _        y )Nr   F)r   r   r   r   )r   r   r   convtr)r   r   r	   r-   r   s       r   r   zConvTrUpsample1d.__init__%   s;    $L.F

r   r   r   c                 &    | j                  ||      S r   )r/   r   s      r   r   zConvTrUpsample1d.forward2   s    {{1k**r   r   r    r*   s   @r   r,   r,       sA    
s 
s 
#* 
+ +D4K +r   r,   )r&   r   pocket_tts.modules.convr   r   Moduler   r,    r   r   <module>r4      s.      M)ryy )2+ryy +r   