o
i, @ sL d Z ddlZddlmZ ddlmZ G dd dejZG dd d ejZdS )
a
This file is part of ComfyUI.
Copyright (C) 2024 Stability AI
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see .
N)nn )LayerNorm2d_opc s& e Zd Zd fdd Zdd Z ZS ) CNetResBlockNc
sd t tt||||dt |j||dddt||||dt |j||ddd| _d S )N)dtypedevice r ) kernel_sizepadding)super__init__r
Sequentialr GELUConv2dblocks)selfcr r
operations __class__ :/mnt/c/Users/fbmor/ComfyUI/comfy/ldm/cascade/controlnet.pyr s
zCNetResBlock.__init__c C s || | S N)r )r xr r r forward$ s zCNetResBlock.forward)NNN)__name__
__module__ __qualname__r r
__classcell__r r r r r s r c s4 e Zd Zddddddef fdd Zdd Z ZS )
ControlNetr i Nc s0 t |d u r d}|| _|dkrod}tj j | _|dkrn| jd d j j
} j |dddd d| jd d< |dkr[| d d d df | jd d j j
d d d df< n| d d d |f | jd d j _
nn|d kr|}t
j ||d
dd d t
jd
ddj |d
|dd d | _nD|dkrt
jj |dd dt
jd
ddj ddd dg fddtdD j ddd dR | _d}ntd| t
| _tt|D ]$}
| jt
j ||d d dt
jd
ddj ||d d d qd| _|| _d| _d S )Neffneti r r F)r stridebiasr r simple r )r r
r r g?T)inplacelargei @ )r r r c s g | ]
}t d dqS )r) )r r r )r .0_r r r r r
E s z'ControlNet.__init__.. zUnknown bottleneck mode: )r r$ r r )r r proj_blocks torchvisionmodelsefficientnet_v2_sfeaturesevalbackboneweightdatar cloner r
LeakyReLUrange
ValueError
ModuleList projectionslenappendxlinput_channelsunshuffle_amount) r c_inc_projr0 bottleneck_moder r r
embd_channels
in_weightsr, r r- r r ) sV
"8&
zControlNet.__init__c C s^ | |}dd tt| jd D }t| jD ]
\}}| j| |||< qd|d d d iS )Nc S s g | ]}d qS r r r* r r r r. Y s z&ControlNet.forward..r input)r6 r; maxr0 enumerater> )r r proj_outputsiidxr r r r W s
zControlNet.forward)r r r r r r r r r r r r ( s .r ) __doc__r1 torchr commonr Moduler r r r r r s