Batter way to write binarize() function. (huggingface#6394)

hisushanta · sayakpaul · yiyixuxu · web-flow · commit 2993257f2aa3 · 2024-01-04T09:32:08.000+05:30
* 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 &lt;spsayakpaul@gmail.com&gt;

* Update src/diffusers/models/unet_2d_blocks.py

Add suggested text

Co-authored-by: Sayak Paul &lt;spsayakpaul@gmail.com&gt;

* 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 &lt;spsayakpaul@gmail.com&gt;
Co-authored-by: YiYi Xu &lt;yixu310@gmail.com&gt;
diff --git a/src/diffusers/image_processor.py b/src/diffusers/image_processor.py
@@ -33,14 +33,7 @@
     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 @@ def convert_to_grayscale(image: PIL.Image.Image) -> PIL.Image.Image:
     @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 @@ def binarize(self, image: PIL.Image.Image) -> PIL.Image.Image:
         """
         image[image < 0.5] = 0
         image[image >= 0.5] = 1
+
         return image
 
     def get_default_height_width(
