or (C, H, W). If it is a NumPy array or
a list of arrays, the expected shape should be (B, H, W, C) or (H, W, C). It can also accept image
latents as image, but if passing latents directly it is not encoded again.   mask_image (torch.FloatTensor, PIL.Image.Image, np.ndarray, List[torch.FloatTensor], —
List[PIL.Image.Image], or List[np.ndarray]):
Image, NumPy array or tensor representing an image batch to mask image. White pixels in the mask
are