code: def setup(self):
      self.conv_in = nn.Conv(
          self.block_out_channels[0],
          kernel_size=(3, 3),
          padding=((1, 1), (1, 1)),
          dtype=self.dtype,
      )
  
      blocks = []
      for i in range(len(self.block_out_channels) - 1):
          channel_in = self.block_out_channels[i]
          channel_out = self.block_out_channels[i + 1]
          conv1 = nn.Conv(
              channel_in,
              kernel_size=(3, 3),
              padding=((1, 1), (1, 1)),
              dtype=self.dtype,
          )
          blocks.append(conv1)
          conv2 = nn.Conv(
              channel_out,
              kernel_size=(3, 3),
              strides=(2, 2),
              padding=((1, 1), (1, 1)),
              dtype=self.dtype,
          )
          blocks.append(conv2)
      self.blocks = blocks
  
      self.conv_out = nn.Conv(
          self.conditioning_embedding_channels,
          kernel_size=(3, 3),
          padding=((1, 1), (1, 1)),
          kernel_init=nn.initializers.zeros_init(),
          bias_init=nn.initializers.zeros_init(),
          dtype=self.dtype,
      )
  
imports: from typing import Optional, Tuple, Union
  import flax
  import flax.linen
  import jax
  import jax.numpy
  from flax.core.frozen_dict import FrozenDict
  from ..configuration_utils import ConfigMixin, flax_register_to_config
  from ..utils import BaseOutput
  from .embeddings_flax import FlaxTimestepEmbedding, FlaxTimesteps
  from .modeling_flax_utils import FlaxModelMixin
  from .unet_2d_blocks_flax import FlaxCrossAttnDownBlock2D, FlaxDownBlock2D, FlaxUNetMidBlock2DCrossAttn
label_desc: Use image node_id 1 for calling the FlaxControlNetConditioningEmbedding obj's underlying member method code with example usage: obj.setup() without return types