s 

unet (UNet2DConditionModel) —
UNet model which weights are copied to the ControlNet. Note that all configuration options are also
copied where applicable.



Instantiate Controlnet class from UNet2DConditionModel.

set_attention_slice

<
source
>
(
slice_size

)


Parameters 

slice_size (str or int or list(int), optional, defaults to "auto") —
When "auto", halves the input to the attention heads, so attention will be computed in two steps.