Ruff format

Author: StAlKeR7779

invokeai/backend/stable_diffusion/extensions/inpaint.py

@@ -17,6 +17,7 @@ class InpaintExt(ExtensionBase):
     """An extension for inpainting with non-inpainting models. See `InpaintModelExt` for inpainting with inpainting
     models.
     """
+
     def __init__(
         self,
         mask: torch.Tensor,
@@ -42,7 +43,7 @@ def __init__(
 
     @staticmethod
     def _is_normal_model(unet: UNet2DConditionModel):
-        """ Checks if the provided UNet belongs to a regular model.
+        """Checks if the provided UNet belongs to a regular model.
         The `in_channels` of a UNet vary depending on model type:
         - normal - 4
         - depth - 5

invokeai/backend/stable_diffusion/extensions/inpaint_model.py

@@ -16,6 +16,7 @@ class InpaintModelExt(ExtensionBase):
     """An extension for inpainting with inpainting models. See `InpaintExt` for inpainting with non-inpainting
     models.
     """
+
     def __init__(
         self,
         mask: Optional[torch.Tensor],
@@ -46,7 +47,7 @@ def __init__(
 
     @staticmethod
     def _is_inpaint_model(unet: UNet2DConditionModel):
-        """ Checks if the provided UNet belongs to a regular model.
+        """Checks if the provided UNet belongs to a regular model.
         The `in_channels` of a UNet vary depending on model type:
         - normal - 4
         - depth - 5