fix: load_image should support PIL Image (#6904)

Laisky · sayakpaul · web-flow · commit 1824d0050e89 · 2024-02-08T08:25:37.000-10:00
Co-authored-by: Sayak Paul &lt;spsayakpaul@gmail.com&gt;
diff --git a/src/diffusers/utils/loading_utils.py b/src/diffusers/utils/loading_utils.py
@@ -32,6 +32,8 @@ def load_image(
             raise ValueError(
                 f"Incorrect path or URL. URLs must start with `http://` or `https://`, and {image} is not a valid path."
             )
+    elif isinstance(image, PIL.Image.Image):
+        image = image
     else:
         raise ValueError(
             "Incorrect format used for the image. Should be a URL linking to an image, a local path, or a PIL image."

Original file line number	Diff line number	Diff line change
`@@ -32,6 +32,8 @@ def load_image(`
`32`	`32`	`raise ValueError(`
`33`	`33`	f"Incorrect path or URL. URLs must start with `http://` or `https://`, and {image} is not a valid path."
`34`	`34`	`)`
	`35`	`+ elif isinstance(image, PIL.Image.Image):`
	`36`	`+ image = image`
`35`	`37`	`else:`
`36`	`38`	`raise ValueError(`
`37`	`39`	`"Incorrect format used for the image. Should be a URL linking to an image, a local path, or a PIL image."`