class_docstr: A text2im model which can perform inpainting.
code: def forward(
      self,
      x,
      timesteps,
      inpaint_image=None,
      inpaint_mask=None,
      **kwargs,
  ):
      if inpaint_image is None:
          inpaint_image = th.zeros_like(x)
      if inpaint_mask is None:
          inpaint_mask = th.zeros_like(x[:, :1])
      return super().forward(
          th.cat(
              [x, inpaint_image * inpaint_mask, inpaint_mask], dim=1
          ),
          timesteps,
          **kwargs,
      )
  
imports: import torch
  import torch.nn
  import torch.nn.functional
  from .nn import timestep_embedding
  from .unet import UNetModel
  from .xf import LayerNorm, Transformer, convert_module_to_f16
label_desc: Use image node_id 2 for calling the InpaintText2ImUNet obj's underlying member method code with example usage: obj.forward(x, timesteps, inpaint_image, inpaint_mask) and returns: super