nd pytorch tensor, the expected value range is between [0, 1] If it’s a tensor or a list
or tensors, the 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[