AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Support grayscale images in numpy_to_pil (#1025)

Browse Source
This commit is contained in:
Anton Lozhkov
2022-10-27 22:44:35 +02:00
committed by GitHub
parent de00c63217
commit fb38bb1621
2 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/diffusers/pipeline_flax_utils.py
View File

@@ -444,7 +444,11 @@ class FlaxDiffusionPipeline(ConfigMixin):
if images.ndim == 3:
images = images[None, ...]
images = (images * 255).round().astype("uint8")
pil_images = [Image.fromarray(image) for image in images]
if images.shape[-1] == 1:
# special case for grayscale (single channel) images
pil_images = [Image.fromarray(image.squeeze(), mode="L") for image in images]
else:
pil_images = [Image.fromarray(image) for image in images]
return pil_images

6
src/diffusers/pipeline_utils.py
View File

@@ -625,7 +625,11 @@ class DiffusionPipeline(ConfigMixin):
if images.ndim == 3:
images = images[None, ...]
images = (images * 255).round().astype("uint8")
pil_images = [Image.fromarray(image) for image in images]
if images.shape[-1] == 1:
# special case for grayscale (single channel) images
pil_images = [Image.fromarray(image.squeeze(), mode="L") for image in images]
else:
pil_images = [Image.fromarray(image) for image in images]
return pil_images