From 4f26c8629bc3f975b15a7766ac2ad494d4b0e0eb Mon Sep 17 00:00:00 2001 From: ddPn08 Date: Fri, 7 Jul 2023 10:54:46 +0900 Subject: [PATCH] format code --- modules/diffusion/networks/lyco.py | 5 +++-- modules/diffusion/pipelines/diffusers.py | 4 ++-- modules/tabs/generate.py | 3 ++- webui.py | 10 +++------- 4 files changed, 10 insertions(+), 12 deletions(-) diff --git a/modules/diffusion/networks/lyco.py b/modules/diffusion/networks/lyco.py index 40b4ef3..f570090 100644 --- a/modules/diffusion/networks/lyco.py +++ b/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") diff --git a/modules/diffusion/pipelines/diffusers.py b/modules/diffusion/pipelines/diffusers.py index 6878805..80d31a3 100644 --- a/modules/diffusion/pipelines/diffusers.py +++ b/modules/diffusion/pipelines/diffusers.py @@ -1,5 +1,4 @@ import copy -from packaging.version import Version import gc import inspect import os @@ -12,8 +11,8 @@ from diffusers import ( AutoencoderKL, DDPMScheduler, - StableDiffusionPipeline, DiffusionPipeline, + StableDiffusionPipeline, UNet2DConditionModel, ) from diffusers.pipelines.stable_diffusion import ( @@ -21,6 +20,7 @@ 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 diff --git a/modules/tabs/generate.py b/modules/tabs/generate.py index 80dbe97..6544a54 100644 --- a/modules/tabs/generate.py +++ b/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 diff --git a/webui.py b/webui.py index 8409070..4f4471f 100644 --- a/webui.py +++ b/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