From 64b36b03125f2ad881a0ea8e923c6abfc9dcc92f Mon Sep 17 00:00:00 2001 From: Edward Li Date: Mon, 11 Dec 2023 03:52:04 -0600 Subject: [PATCH] Correct type annotation for `VaeImageProcessor.numpy_to_pil` (#6111) From `(np.ndarray) -> PIL.Image.Image` to `(np.ndarray) -> List[PIL.Image.Image]`. --- src/diffusers/image_processor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/image_processor.py b/src/diffusers/image_processor.py index a515805fd0871..ab96384fe9f1a 100644 --- a/src/diffusers/image_processor.py +++ b/src/diffusers/image_processor.py @@ -88,7 +88,7 @@ def __init__( self.config.do_convert_rgb = False @staticmethod - def numpy_to_pil(images: np.ndarray) -> PIL.Image.Image: + def numpy_to_pil(images: np.ndarray) -> List[PIL.Image.Image]: """ Convert a numpy image or a batch of images to a PIL image. """