
expected shape should be (B, C, H, W) 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 mas