chore: ruff

invokeai/backend/image_util/__init__.py

@@ -1,6 +1,7 @@
 """
 Initialization file for invokeai.backend.image_util methods.
 """
+
 from .patchmatch import PatchMatch  # noqa: F401
 from .pngwriter import PngWriter, PromptFormatter, retrieve_metadata, write_metadata  # noqa: F401
 from .seamless import configure_model_padding  # noqa: F401

invokeai/backend/image_util/invisible_watermark.py

@@ -3,6 +3,7 @@ This module defines a singleton object, "invisible_watermark" that
 wraps the invisible watermark model. It respects the global "invisible_watermark"
 configuration variable, that allows the watermarking to be supressed.
 """
+
 import cv2
 import numpy as np
 from imwatermark import WatermarkEncoder

invokeai/backend/image_util/patchmatch.py

@@ -4,6 +4,7 @@ wraps the actual patchmatch object. It respects the global
 "try_patchmatch" attribute, so that patchmatch loading can
 be suppressed or deferred
 """
+
 import numpy as np
 
 import invokeai.backend.util.logging as logger

invokeai/backend/image_util/pngwriter.py

@@ -6,6 +6,7 @@ PngWriter -- Converts Images generated by T2I into PNGs, finds
 
 Exports function retrieve_metadata(path)
 """
+
 import json
 import os
 import re

invokeai/backend/image_util/safety_checker.py

@@ -3,6 +3,7 @@ This module defines a singleton object, "safety_checker" that
 wraps the safety_checker model. It respects the global "nsfw_checker"
 configuration variable, that allows the checker to be supressed.
 """
+
 import numpy as np
 from PIL import Image
 

invokeai/backend/install/check_root.py

@@ -1,6 +1,7 @@
 """
 Check that the invokeai_root is correctly configured and exit if not.
 """
+
 import sys
 
 from invokeai.app.services.config import InvokeAIAppConfig

invokeai/backend/install/install_helper.py

@@ -1,4 +1,5 @@
 """Utility (backend) functions used by model_install.py"""
+
 from logging import Logger
 from pathlib import Path
 from typing import Any, Dict, List, Optional

invokeai/backend/model_manager/__init__.py

@@ -1,4 +1,5 @@
 """Re-export frequently-used symbols from the Model Manager backend."""
+
 from .config import (
     AnyModel,
     AnyModelConfig,

invokeai/backend/model_manager/config.py

@@ -19,6 +19,7 @@ Typical usage:
 Validation errors will raise an InvalidModelConfigException error.
 
 """
+
 import time
 from enum import Enum
 from typing import Literal, Optional, Type, Union

invokeai/backend/model_manager/convert_ckpt_to_diffusers.py

@@ -15,7 +15,7 @@
 #
 # Adapted for use in InvokeAI by Lincoln Stein, July 2023
 #
-""" Conversion script for the Stable Diffusion checkpoints."""
+"""Conversion script for the Stable Diffusion checkpoints."""
 
 import re
 from contextlib import nullcontext

invokeai/backend/model_manager/load/__init__.py

@@ -2,6 +2,7 @@
 """
 Init file for the model loader.
 """
+
 from importlib import import_module
 from pathlib import Path
 

invokeai/backend/model_manager/load/convert_cache/convert_cache_base.py

@@ -1,6 +1,7 @@
 """
 Disk-based converted model cache.
 """
+
 from abc import ABC, abstractmethod
 from pathlib import Path
 

invokeai/backend/model_manager/load/model_loader_registry.py

@@ -14,6 +14,7 @@ Use like this:
     ).load_model(model_config, submodel_type)
 
 """
+
 import hashlib
 from abc import ABC, abstractmethod
 from pathlib import Path

invokeai/backend/model_manager/load/model_loaders/lora.py

@@ -1,7 +1,6 @@
 # Copyright (c) 2024, Lincoln D. Stein and the InvokeAI Development Team
 """Class for LoRA model loading in InvokeAI."""
 
-
 from logging import Logger
 from pathlib import Path
 from typing import Optional, Tuple

invokeai/backend/model_manager/load/model_loaders/stable_diffusion.py

@@ -1,7 +1,6 @@
 # Copyright (c) 2024, Lincoln D. Stein and the InvokeAI Development Team
 """Class for StableDiffusion model loading in InvokeAI."""
 
-
 from pathlib import Path
 from typing import Optional
 

invokeai/backend/model_manager/load/model_loaders/textual_inversion.py

@@ -1,7 +1,6 @@
 # Copyright (c) 2024, Lincoln D. Stein and the InvokeAI Development Team
 """Class for TI model loading in InvokeAI."""
 
-
 from pathlib import Path
 from typing import Optional, Tuple
 

invokeai/backend/model_manager/metadata/__init__.py

@@ -18,6 +18,7 @@ assert isinstance(data, CivitaiMetadata)
 if data.allow_commercial_use:
    print("Commercial use of this model is allowed")
 """
+
 from .fetch import CivitaiMetadataFetch, HuggingFaceMetadataFetch, ModelMetadataFetchBase
 from .metadata_base import (
     AnyModelRepoMetadata,

invokeai/backend/model_patcher.py

@@ -1,5 +1,6 @@
 # Copyright (c) 2024 Ryan Dick, Lincoln D. Stein, and the InvokeAI Development Team
 """These classes implement model patching with LoRAs and Textual Inversions."""
+
 from __future__ import annotations
 
 import pickle

invokeai/backend/stable_diffusion/__init__.py

@@ -1,6 +1,7 @@
 """
 Initialization file for the invokeai.backend.stable_diffusion package
 """
+
 from .diffusers_pipeline import PipelineIntermediateState, StableDiffusionGeneratorPipeline  # noqa: F401
 from .diffusion import InvokeAIDiffuserComponent  # noqa: F401
 from .diffusion.cross_attention_map_saving import AttentionMapSaver  # noqa: F401

invokeai/backend/stable_diffusion/diffusion/__init__.py

@@ -1,6 +1,7 @@
 """
 Initialization file for invokeai.models.diffusion
 """
+
 from .cross_attention_control import InvokeAICrossAttentionMixin  # noqa: F401
 from .cross_attention_map_saving import AttentionMapSaver  # noqa: F401
 from .shared_invokeai_diffusion import InvokeAIDiffuserComponent  # noqa: F401

invokeai/backend/training/__init__.py

@@ -1,4 +1,5 @@
 """
 Initialization file for invokeai.backend.training
 """
+
 from .textual_inversion_training import do_textual_inversion_training, parse_args  # noqa: F401

invokeai/backend/training/textual_inversion_training.py

@@ -858,9 +858,9 @@ def do_textual_inversion_training(
                 # Let's make sure we don't update any embedding weights besides the newly added token
                 index_no_updates = torch.arange(len(tokenizer)) != placeholder_token_id
                 with torch.no_grad():
-                    accelerator.unwrap_model(text_encoder).get_input_embeddings().weight[
-                        index_no_updates
-                    ] = orig_embeds_params[index_no_updates]
+                    accelerator.unwrap_model(text_encoder).get_input_embeddings().weight[index_no_updates] = (
+                        orig_embeds_params[index_no_updates]
+                    )
 
             # Checks if the accelerator has performed an optimization step behind the scenes
             if accelerator.sync_gradients:

invokeai/backend/util/__init__.py

@@ -1,6 +1,7 @@
 """
 Initialization file for invokeai.backend.util
 """
+
 from .attention import auto_detect_slice_size  # noqa: F401
 from .devices import (  # noqa: F401
     CPU_DEVICE,

invokeai/backend/util/attention.py

@@ -3,6 +3,7 @@
 Utility routine used for autodetection of optimal slice size
 for attention mechanism.
 """
+
 import psutil
 import torch
 

invokeai/backend/util/silence_warnings.py

@@ -1,4 +1,5 @@
 """Context class to silence transformers and diffusers warnings."""
+
 import warnings
 from typing import Any
 

invokeai/backend/util/util.py

@@ -340,14 +340,17 @@ def download_with_resume(url: str, dest: Path, access_token: str = None) -> Path
             logger.error(f"ERROR DOWNLOADING {url}: {resp.text}")
             return None
 
-        with open(dest, open_mode) as file, tqdm(
-            desc=str(dest),
-            initial=exist_size,
-            total=content_length,
-            unit="iB",
-            unit_scale=True,
-            unit_divisor=1000,
-        ) as bar:
+        with (
+            open(dest, open_mode) as file,
+            tqdm(
+                desc=str(dest),
+                initial=exist_size,
+                total=content_length,
+                unit="iB",
+                unit_scale=True,
+                unit_divisor=1000,
+            ) as bar,
+        ):
             for data in resp.iter_content(chunk_size=1024):
                 size = file.write(data)
                 bar.update(size)