From 2993257f2aa30290fea00e698e8ae9285ee65255 Mon Sep 17 00:00:00 2001 From: Chi Date: Thu, 4 Jan 2024 09:32:08 +0530 Subject: [PATCH] Batter way to write binarize() function. (#6394) * I added a new doc string to the class. This is more flexible to understanding other developers what are doing and where it's using. * Update src/diffusers/models/unet_2d_blocks.py This changes suggest by maintener. Co-authored-by: Sayak Paul * Update src/diffusers/models/unet_2d_blocks.py Add suggested text Co-authored-by: Sayak Paul * Update unet_2d_blocks.py I changed the Parameter to Args text. * Update unet_2d_blocks.py proper indentation set in this file. * Update unet_2d_blocks.py a little bit of change in the act_fun argument line. * I run the black command to reformat style in the code * Update unet_2d_blocks.py similar doc-string add to have in the original diffusion repository. * Batter way to write binarize function * Solve check_code_quality error * My mistake to run pull request but not reformated file * Update image_processor.py * remove extra variable and space * Update image_processor.py * Run ruff libarary to reformat my file --------- Co-authored-by: Sayak Paul Co-authored-by: YiYi Xu --- src/diffusers/image_processor.py | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/src/diffusers/image_processor.py b/src/diffusers/image_processor.py index 9893f71638..843052c1ad 100644 --- a/src/diffusers/image_processor.py +++ b/src/diffusers/image_processor.py @@ -33,14 +33,7 @@ PipelineImageInput = Union[ List[torch.FloatTensor], ] -PipelineDepthInput = Union[ - PIL.Image.Image, - np.ndarray, - torch.FloatTensor, - List[PIL.Image.Image], - List[np.ndarray], - List[torch.FloatTensor], -] +PipelineDepthInput = PipelineImageInput class VaeImageProcessor(ConfigMixin): @@ -169,7 +162,7 @@ class VaeImageProcessor(ConfigMixin): @staticmethod def blur(image: PIL.Image.Image, blur_factor: int = 4) -> PIL.Image.Image: """ - Blurs an image. + Applies Gaussian blur to an image. """ image = image.filter(ImageFilter.GaussianBlur(blur_factor)) @@ -402,6 +395,7 @@ class VaeImageProcessor(ConfigMixin): """ image[image < 0.5] = 0 image[image >= 0.5] = 1 + return image def get_default_height_width(