ge.Image) —
Image, or tensor representing an image batch, to mask image. White pixels in the mask will be
repainted, while black pixels will be preserved. If mask_image is a PIL image, it will be converted
to a single channel (luminance) before use. If it’s a tensor, it should contain one color channel (L)
instead of 3, so the expected shape would be (B, H, W, 1).   height (int, optional, defaults to self.unet.config.sample_size * self.vae_scal