Flux2KleinPipeline doesn't accept tensor format as input image

[joansc]

Is your feature request related to a problem? Please describe.

I would like to input a tensor in the Flux2KleinPipeline image argument. Only image: list[PIL.Image.Image] | PIL.Image.Image | None = None is accepted although in the comment section below the arguments it states that:

image (torch.Tensor, PIL.Image.Image, np.ndarray, List[torch.Tensor], List[PIL.Image.Image], or List[np.ndarray]): Image, numpy array or tensor representing an image batch to be used as the starting point. For both numpy array and 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 asimage, but if passing latents directly it is not encoded again.

Thanks in advance,

Joan