format code

modules/diffusion/networks/lyco.py

@@ -1,8 +1,9 @@
 import os
+
+import safetensors.torch
+import torch
 from lycoris import kohya
 from lycoris.modules import locon, loha
-import torch
-import safetensors.torch
 
 kohya.LycorisNetwork.UNET_TARGET_REPLACE_MODULE.remove("Attention")
 

modules/diffusion/pipelines/diffusers.py

@@ -1,5 +1,4 @@
 import copy
-from packaging.version import Version
 import gc
 import inspect
 import os
@@ -12,15 +11,16 @@
 from diffusers import (
     AutoencoderKL,
     DDPMScheduler,
-    StableDiffusionPipeline,
     DiffusionPipeline,
+    StableDiffusionPipeline,
     UNet2DConditionModel,
 )
 from diffusers.pipelines.stable_diffusion import (
     StableDiffusionPipelineOutput,
     convert_from_ckpt,
 )
 from diffusers.utils import PIL_INTERPOLATION, numpy_to_pil, randn_tensor
+from packaging.version import Version
 from safetensors.torch import load_file
 from tqdm import tqdm
 from transformers import CLIPTextModel, CLIPTokenizer

modules/tabs/generate.py

@@ -2,10 +2,11 @@
 from typing import *
 
 import gradio as gr
+
 from api.models.diffusion import (
     HiresfixOptions,
-    MultidiffusionOptions,
     ImageGenerationOptions,
+    MultidiffusionOptions,
 )
 from modules import model_manager
 from modules.components import gallery, image_generation_options

webui.py

@@ -11,13 +11,9 @@
     print(f"TensorRT version: {trt.__version__}")
     trt.init_libnvinfer_plugins(TRT_LOGGER, "")
 
-from api.events.common import (
-    PostAppLaunchEvent,
-    PostUICreateEvent,
-    PreAppLaunchEvent,
-    PreUICreateEvent,
-)
-from modules import config, model_manager, plugin_loader, ui, javascripts
+from api.events.common import (PostAppLaunchEvent, PostUICreateEvent,
+                               PreAppLaunchEvent, PreUICreateEvent)
+from modules import config, javascripts, model_manager, plugin_loader, ui
 from modules.diffusion import embeddings, networks