
    %h                         d dl Zd dlZd dlmZ d dlmc mZ d dlm	Z	 d dl
mZ  e       Z
 G d dej                        Zy)    N)partial)Configc                   &     e Zd Zd fd	Zd Z xZS )BasicLatBlkc                     t         t        |           t        j                  dk(  r|dz  nd}t        j                  ||ddd      | _        y )Nadap   @      r   )superr   __init__configdec_channels_internnConv2dconv)selfin_channelsout_channelsinter_channels	__class__s       ?/var/www/html/mariraj/BiRefNet/models/modules/lateral_blocks.pyr   zBasicLatBlk.__init__   sB    k4)+-3-F-F&-P)VXIIk<AqA	    c                 (    | j                  |      }|S )N)r   )r   xs     r   forwardzBasicLatBlk.forward   s    IIaLr   )r
   r
   r
   )__name__
__module____qualname__r   r   __classcell__)r   s   @r   r   r      s    B
r   r   )numpynptorchtorch.nnr   torch.nn.functional
functionalF	functoolsr   r   r   Moduler    r   r   <module>r+      s3           
")) r   