Skip to content

Commit

Permalink
Correct type annotation for VaeImageProcessor.numpy_to_pil (hugging…
Browse files Browse the repository at this point in the history
…face#6111)

From `(np.ndarray) -> PIL.Image.Image` to `(np.ndarray) -> List[PIL.Image.Image]`.
  • Loading branch information
edwardwli authored and Jimmy committed Apr 26, 2024
1 parent c119914 commit 64b36b0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/diffusers/image_processor.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
"""
Expand Down

0 comments on commit 64b36b0

Please sign in to comment.