chore: ruff

invokeai/app/services/config/config_default.py

@@ -166,6 +166,7 @@ two configs are kept in separate sections of the config file:
      ...
 
 """
+
 from __future__ import annotations
 
 import os

invokeai/app/services/download/__init__.py

@@ -1,4 +1,5 @@
 """Init file for download queue."""
+
 from .download_base import DownloadJob, DownloadJobStatus, DownloadQueueServiceBase, UnknownJobIDException
 from .download_default import DownloadQueueService, TqdmProgress
 

invokeai/app/services/model_records/__init__.py

@@ -1,4 +1,5 @@
 """Init file for model record services."""
+
 from .model_records_base import (  # noqa F401
     DuplicateModelException,
     InvalidModelException,

invokeai/app/services/model_records/model_records_sql.py

@@ -39,7 +39,6 @@ Typical usage:
   configs = store.search_by_attr(base_model='sd-2', model_type='main')
 """
 
-
 import json
 import sqlite3
 from math import ceil

invokeai/app/services/shared/sqlite_migrator/sqlite_migrator_common.py

@@ -17,8 +17,7 @@ class MigrateCallback(Protocol):
     See :class:`Migration` for an example.
     """
 
-    def __call__(self, cursor: sqlite3.Cursor) -> None:
+    def __call__(self, cursor: sqlite3.Cursor) -> None: ...
-        ...
 
 
 class MigrationError(RuntimeError):

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[
+                    accelerator.unwrap_model(text_encoder).get_input_embeddings().weight[index_no_updates] = (
-                        index_no_updates
+                        orig_embeds_params[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(
+        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:
+            ) as bar,
+        ):
             for data in resp.iter_content(chunk_size=1024):
                 size = file.write(data)
                 bar.update(size)

invokeai/frontend/CLI/__init__.py

@@ -1,4 +1,5 @@
 """
 Initialization file for invokeai.frontend.CLI
 """
+
 from .CLI import main as invokeai_command_line_interface  # noqa: F401

invokeai/frontend/install/invokeai_configure.py

@@ -1,6 +1,7 @@
 """
 Wrapper for invokeai.backend.configure.invokeai_configure
 """
+
 from ...backend.install.invokeai_configure import main as invokeai_configure  # noqa: F401
 
 __all__ = ["invokeai_configure"]

invokeai/frontend/install/invokeai_update.py

@@ -2,6 +2,7 @@
 Minimalist updater script. Prompts user for the tag or branch to update to and runs
 pip install <path_to_git_source>.
 """
+
 import os
 import platform
 from distutils.version import LooseVersion

invokeai/frontend/install/widgets.py

@@ -1,6 +1,7 @@
 """
 Widget class definitions used by model_select.py, merge_diffusers.py and textual_inversion.py
 """
+
 import curses
 import math
 import os

invokeai/frontend/merge/__init__.py

@@ -1,4 +1,5 @@
 """
 Initialization file for invokeai.frontend.merge
 """
+
 from .merge_diffusers import main as invokeai_merge_diffusers  # noqa: F401

invokeai/frontend/merge/merge_diffusers.py

@@ -4,6 +4,7 @@ used to merge 2-3 models together and create a new InvokeAI-registered diffusion
 
 Copyright (c) 2023 Lincoln Stein and the InvokeAI Development Team
 """
+
 import argparse
 import curses
 import re

invokeai/frontend/training/__init__.py

@@ -1,4 +1,5 @@
 """
 Initialization file for invokeai.frontend.training
 """
+
 from .textual_inversion import main as invokeai_textual_inversion  # noqa: F401

invokeai/frontend/training/textual_inversion.py

@@ -6,7 +6,6 @@ This is the frontend to "textual_inversion_training.py".
 Copyright (c) 2023-24 Lincoln Stein and the InvokeAI Development Team
 """
 
-
 import os
 import re
 import shutil

invokeai/version/__init__.py

@@ -1,6 +1,7 @@
 """
 initialization file for invokeai
 """
+
 from .invokeai_version import __version__  # noqa: F401
 
 __app_id__ = "invoke-ai/InvokeAI"

tests/app/services/download/test_download_queue.py

@@ -1,4 +1,5 @@
 """Test the queued download facility"""
+
 import re
 import time
 from pathlib import Path

tests/backend/model_manager/model_metadata/test_model_metadata.py

@@ -1,6 +1,7 @@
 """
 Test model metadata fetching and storage.
 """
+
 import datetime
 from pathlib import Path
 

tests/backend/util/test_logging.py

@@ -1,6 +1,7 @@
 """
 Test interaction of logging with configuration system.
 """
+
 import io
 import logging
 import re

tests/test_path.py

@@ -2,6 +2,7 @@
 Not really a test, but a way to verify that the paths are existing
 and fail early if they are not.
 """
+
 import pathlib
 import unittest
 from os import path as osp