Revert "Fix ruff linter (#137)" (#143)

This reverts commit 6b3ad64388.

modules/initialize.py

@@ -3,6 +3,7 @@ import logging
 import os
 import sys
 import warnings
+import os
 
 from modules.timer import startup_timer
 
@@ -37,7 +38,7 @@ def imports():
         startup_timer.record("setup paths")
 
         import ldm.modules.encoders.modules  # noqa: F401
-        import ldm.modules.diffusionmodules.model  # noqa: F401
+        import ldm.modules.diffusionmodules.model
         startup_timer.record("import ldm")
 
         import sgm.modules.encoders.modules  # noqa: F401

modules/lowvram.py

@@ -1,5 +1,5 @@
 import torch
-
+from modules import devices, shared
 
 module_in_gpu = None
 cpu = torch.device("cpu")

modules/paths.py

@@ -3,7 +3,6 @@ import sys
 from modules.paths_internal import models_path, script_path, data_path, extensions_dir, extensions_builtin_dir, cwd  # noqa: F401
 
 import modules.safe  # noqa: F401
-import ldm_patched.utils.path_utils as ldm_patched_path_utils
 
 
 def mute_sdxl_imports():
@@ -65,6 +64,8 @@ for d, must_exist, what, options in path_dirs:
         paths[what] = d
 
 
+import ldm_patched.utils.path_utils as ldm_patched_path_utils
+
 ldm_patched_path_utils.base_path = data_path
 ldm_patched_path_utils.models_dir = models_path
 ldm_patched_path_utils.output_directory = os.path.join(data_path, "output")

modules/processing.py

@@ -16,7 +16,7 @@ from skimage import exposure
 from typing import Any
 
 import modules.sd_hijack
-from modules import devices, prompt_parser, masking, sd_samplers, infotext_utils, extra_networks, sd_vae_approx, scripts, sd_samplers_common, sd_unet, rng
+from modules import devices, prompt_parser, masking, sd_samplers, lowvram, infotext_utils, extra_networks, sd_vae_approx, scripts, sd_samplers_common, sd_unet, errors, rng
 from modules.rng import slerp # noqa: F401
 from modules.sd_hijack import model_hijack
 from modules.sd_samplers_common import images_tensor_to_samples, decode_first_stage, approximation_indexes
@@ -517,9 +517,9 @@ class StableDiffusionProcessing:
 
 
 class Processed:
-    def __init__(self, p: StableDiffusionProcessing, images_list, seed=-1, info="", subseed=None, all_prompts=None, all_negative_prompts=None, all_seeds=None, all_subseeds=None, index_of_first_image=0, infotexts=None, comments="", extra_images_list=None):
+    def __init__(self, p: StableDiffusionProcessing, images_list, seed=-1, info="", subseed=None, all_prompts=None, all_negative_prompts=None, all_seeds=None, all_subseeds=None, index_of_first_image=0, infotexts=None, comments="", extra_images_list=[]):
         self.images = images_list
-        self.extra_images = extra_images_list if extra_images_list is not None else []
+        self.extra_images = extra_images_list
         self.prompt = p.prompt
         self.negative_prompt = p.negative_prompt
         self.seed = seed

modules/sd_hijack.py

@@ -1,8 +1,11 @@
 import torch
+from torch.nn.functional import silu
 from types import MethodType
 
-from modules import devices, sd_hijack_optimizations, shared, script_callbacks, sd_unet, patches
+from modules import devices, sd_hijack_optimizations, shared, script_callbacks, errors, sd_unet, patches
+from modules.hypernetworks import hypernetwork
 from modules.shared import cmd_opts
+from modules import sd_hijack_clip, sd_hijack_open_clip, sd_hijack_unet, sd_hijack_xlmr, xlmr, xlmr_m18
 
 import ldm.modules.attention
 import ldm.modules.diffusionmodules.model

modules/sd_models.py

@@ -6,17 +6,22 @@ import threading
 import torch
 import re
 import safetensors.torch
-from omegaconf import ListConfig
+from omegaconf import OmegaConf, ListConfig
 from os import mkdir
 from urllib import request
 import ldm.modules.midas as midas
 import gc
 
-from modules import paths, shared, modelloader, devices, script_callbacks, sd_vae, errors, hashes, cache, extra_networks, processing, patches
+from ldm.util import instantiate_from_config
+
+from modules import paths, shared, modelloader, devices, script_callbacks, sd_vae, sd_disable_initialization, errors, hashes, sd_models_config, sd_unet, sd_models_xl, cache, extra_networks, processing, lowvram, sd_hijack, patches
 from modules.timer import Timer
 import numpy as np
 from modules_forge import forge_loader
+import modules_forge.ops as forge_ops
+from ldm_patched.modules.ops import manual_cast
 from ldm_patched.modules import model_management as model_management
+import ldm_patched.modules.model_patcher
 
 
 model_dir = "Stable-diffusion"

modules/sd_samplers_cfg_denoiser.py

@@ -1,9 +1,10 @@
 import torch
-from modules import prompt_parser, sd_samplers_common
+from modules import prompt_parser, devices, sd_samplers_common
 
-from modules.shared import state
+from modules.shared import opts, state
 import modules.shared as shared
 from modules.script_callbacks import CFGDenoiserParams, cfg_denoiser_callback
+from modules.script_callbacks import CFGDenoisedParams, cfg_denoised_callback
 from modules.script_callbacks import AfterCFGCallbackParams, cfg_after_cfg_callback
 from modules_forge import forge_sampler
 

modules/sd_unet.py

@@ -1,6 +1,6 @@
 import torch.nn
 
-from modules import script_callbacks, shared
+from modules import script_callbacks, shared, devices
 
 unet_options = []
 current_unet_option = None

modules/sd_vae.py

@@ -2,7 +2,7 @@ import os
 import collections
 from dataclasses import dataclass
 
-from modules import paths, shared, script_callbacks, sd_models, extra_networks, sd_hijack, hashes
+from modules import paths, shared, devices, script_callbacks, sd_models, extra_networks, lowvram, sd_hijack, hashes
 
 import glob
 from copy import deepcopy

modules/shared_init.py

@@ -1,5 +1,7 @@
 import os
 
+import torch
+
 from modules import shared
 from modules.shared import cmd_opts
 

modules/ui.py

@@ -171,7 +171,7 @@ def update_token_counter(text, steps, *, is_positive=True):
         cond_stage_model = sd_models.model_data.sd_model.cond_stage_model
         assert cond_stage_model is not None
     except Exception:
-        return "<span class='gr-box gr-text-input'>?/?</span>"
+        return f"<span class='gr-box gr-text-input'>?/?</span>"
 
     flat_prompts = reduce(lambda list1, list2: list1+list2, prompt_schedules)
     prompts = [prompt_text for step, prompt_text in flat_prompts]

modules/upscaler.py

@@ -14,7 +14,7 @@ def prepare_free_memory(aggressive=False):
         model_management.unload_all_models()
         print('Upscale script freed all memory.')
         return
-
+    
     model_management.free_memory(memory_required=1024*1024*3, device=model_management.get_torch_device())
     print('Upscale script freed memory successfully.')