Correct type annotation for VaeImageProcessor.numpy_to_pil (hugging…

…face#6111) From `(np.ndarray) -> PIL.Image.Image` to `(np.ndarray) -> List[PIL.Image.Image]`.
cardinalblue · Dec 11, 2023 · 664e931 · 664e931
1 parent 88bdd97
commit 664e931
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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.
         """