    def forward(self, conditioning):
        """
        Padding required!
        """
        pad = ops.nhwc3to4()
        conditioning = pad(conditioning)
        embedding = self.conv_in(conditioning)
        embedding = ops.silu(embedding)

        for block in self.blocks:
            embedding = block(embedding)
            embedding = ops.silu(embedding)

        embedding = self.conv_out(embedding)

        return embedding
Padding required!
examples/05_stable_diffusion/src/modeling/controlnet_unet_2d_condition.py