Reformatting

ldm/models/diffusion/ddim.py

@@ -472,7 +472,6 @@ class DDIMSampler(object):
         use_original_steps=False,
         callback=None,
     ):
-
         timesteps = (
             np.arange(self.ddpm_num_timesteps)
             if use_original_steps

ldm/models/diffusion/ddpm.py

@@ -1358,7 +1358,6 @@ class LatentDiffusion(DDPM):
         start_T=None,
         log_every_t=None,
     ):
-
         if not log_every_t:
             log_every_t = self.log_every_t
         device = self.betas.device

ldm/modules/image_degradation/bsrgan.py

@@ -519,7 +519,6 @@ def degradation_bsrgan(img, sf=4, lq_patchsize=72, isp_model=None):
         )
 
     for i in shuffle_order:
-
         if i == 0:
             img = add_blur(img, sf=sf)
 
@@ -623,7 +622,6 @@ def degradation_bsrgan_variant(image, sf=4, isp_model=None):
         )
 
     for i in shuffle_order:
-
         if i == 0:
             image = add_blur(image, sf=sf)
 

ldm/modules/image_degradation/bsrgan_light.py

@@ -520,7 +520,6 @@ def degradation_bsrgan(img, sf=4, lq_patchsize=72, isp_model=None):
         )
 
     for i in shuffle_order:
-
         if i == 0:
             img = add_blur(img, sf=sf)
 
@@ -624,7 +623,6 @@ def degradation_bsrgan_variant(image, sf=4, isp_model=None, up=False):
         )
 
     for i in shuffle_order:
-
         if i == 0:
             image = add_blur(image, sf=sf)
 

ldm/modules/image_degradation/utils_image.py

@@ -271,22 +271,18 @@ def read_img(path):
 
 
 def uint2single(img):
-
     return np.float32(img / 255.0)
 
 
 def single2uint(img):
-
     return np.uint8((img.clip(0, 1) * 255.0).round())
 
 
 def uint162single(img):
-
     return np.float32(img / 65535.0)
 
 
 def single2uint16(img):
-
     return np.uint16((img.clip(0, 1) * 65535.0).round())
 
 
@@ -586,18 +582,14 @@ def rgb2ycbcr(img, only_y=True):
     if only_y:
         rlt = np.dot(img, [65.481, 128.553, 24.966]) / 255.0 + 16.0
     else:
-        rlt = (
+        rlt = np.matmul(
-            np.matmul(
             img,
             [
                 [65.481, -37.797, 112.0],
                 [128.553, -74.203, -93.786],
                 [24.966, 112.0, -18.214],
             ],
-            )
+        ) / 255.0 + [16, 128, 128]
-            / 255.0
-            + [16, 128, 128]
-        )
     if in_img_type == np.uint8:
         rlt = rlt.round()
     else:
@@ -616,18 +608,14 @@ def ycbcr2rgb(img):
     if in_img_type != np.uint8:
         img *= 255.0
     # convert
-    rlt = (
+    rlt = np.matmul(
-        np.matmul(
         img,
         [
             [0.00456621, 0.00456621, 0.00456621],
             [0, -0.00153632, 0.00791071],
             [0.00625893, -0.00318811, 0],
         ],
-        )
+    ) * 255.0 + [-222.921, 135.576, -276.836]
-        * 255.0
-        + [-222.921, 135.576, -276.836]
-    )
     if in_img_type == np.uint8:
         rlt = rlt.round()
     else:
@@ -650,18 +638,14 @@ def bgr2ycbcr(img, only_y=True):
     if only_y:
         rlt = np.dot(img, [24.966, 128.553, 65.481]) / 255.0 + 16.0
     else:
-        rlt = (
+        rlt = np.matmul(
-            np.matmul(
             img,
             [
                 [24.966, 112.0, -18.214],
                 [128.553, -74.203, -93.786],
                 [65.481, -37.797, 112.0],
             ],
-            )
+        ) / 255.0 + [16, 128, 128]
-            / 255.0
-            + [16, 128, 128]
-        )
     if in_img_type == np.uint8:
         rlt = rlt.round()
     else:

ldm/modules/karlo/kakao/sampler.py

@@ -36,7 +36,6 @@ class T2ISampler(BaseSampler):
         clip_stat_path: str,
         sampling_type: str = "default",
     ):
-
         model = cls(
             root_dir=root_dir,
             sampling_type=sampling_type,

ldm/modules/midas/midas/dpt_depth.py

@@ -33,7 +33,6 @@ class DPT(BaseModel):
         channels_last=False,
         use_bn=False,
     ):
-
         super(DPT, self).__init__()
 
         self.channels_last = channels_last

ldm/modules/midas/utils.py

@@ -17,7 +17,6 @@ def read_pfm(path):
         tuple: (data, scale)
     """
     with open(path, "rb") as file:
-
         color = None
         width = None
         height = None

requirements.txt

@@ -16,7 +16,7 @@ gradio==3.13.2
 kornia==0.6
 invisible-watermark>=0.1.5
 streamlit-drawable-canvas==0.8.0
-black==21.9b0
+black==23.3.0
 isort==5.9.3
 flake8==4.0.1
 click==8.0.3