Unverified Commit 664e931b authored by Edward Li's avatar Edward Li Committed by GitHub
Browse files

Correct type annotation for `VaeImageProcessor.numpy_to_pil` (#6111)

From `(np.ndarray) -> PIL.Image.Image` to `(np.ndarray) -> List[PIL.Image.Image]`.
parent 88bdd97c
...@@ -88,7 +88,7 @@ class VaeImageProcessor(ConfigMixin): ...@@ -88,7 +88,7 @@ class VaeImageProcessor(ConfigMixin):
self.config.do_convert_rgb = False self.config.do_convert_rgb = False
@staticmethod @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. Convert a numpy image or a batch of images to a PIL image.
""" """
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment