From 635d2f480d0210b204e28981a2696fc8da8555d6 Mon Sep 17 00:00:00 2001 From: Ryan Dick Date: Mon, 26 Aug 2024 14:09:02 +0000 Subject: [PATCH] ruff --- invokeai/app/invocations/model.py | 3 +-- invokeai/backend/model_manager/load/model_loaders/flux.py | 2 +- .../quantization/scripts/load_flux_model_bnb_llm_int8.py | 4 ++-- 3 files changed, 4 insertions(+), 5 deletions(-) diff --git a/invokeai/app/invocations/model.py b/invokeai/app/invocations/model.py index d68f8eaa97..88874f302a 100644 --- a/invokeai/app/invocations/model.py +++ b/invokeai/app/invocations/model.py @@ -1,7 +1,6 @@ import copy from typing import List, Literal, Optional -import yaml from pydantic import BaseModel, Field from invokeai.app.invocations.baseinvocation import ( @@ -11,10 +10,10 @@ from invokeai.app.invocations.baseinvocation import ( invocation, invocation_output, ) -from invokeai.backend.flux.util import max_seq_lengths from invokeai.app.invocations.fields import FieldDescriptions, Input, InputField, OutputField, UIType from invokeai.app.services.shared.invocation_context import InvocationContext from invokeai.app.shared.models import FreeUConfig +from invokeai.backend.flux.util import max_seq_lengths from invokeai.backend.model_manager.config import ( AnyModelConfig, BaseModelType, diff --git a/invokeai/backend/model_manager/load/model_loaders/flux.py b/invokeai/backend/model_manager/load/model_loaders/flux.py index 063367f30d..33090ccc10 100644 --- a/invokeai/backend/model_manager/load/model_loaders/flux.py +++ b/invokeai/backend/model_manager/load/model_loaders/flux.py @@ -11,8 +11,8 @@ from transformers import AutoConfig, AutoModelForTextEncoding, CLIPTextModel, CL from invokeai.app.services.config.config_default import get_config from invokeai.backend.flux.model import Flux -from invokeai.backend.flux.util import configs, ae_params from invokeai.backend.flux.modules.autoencoder import AutoEncoder +from invokeai.backend.flux.util import ae_params, configs from invokeai.backend.model_manager import ( AnyModel, AnyModelConfig, diff --git a/invokeai/backend/quantization/scripts/load_flux_model_bnb_llm_int8.py b/invokeai/backend/quantization/scripts/load_flux_model_bnb_llm_int8.py index 286c96b527..51c787d8ef 100644 --- a/invokeai/backend/quantization/scripts/load_flux_model_bnb_llm_int8.py +++ b/invokeai/backend/quantization/scripts/load_flux_model_bnb_llm_int8.py @@ -1,10 +1,10 @@ from pathlib import Path import accelerate -from invokeai.backend.flux.model import Flux -from invokeai.backend.flux.util import configs as flux_configs from safetensors.torch import load_file, save_file +from invokeai.backend.flux.model import Flux +from invokeai.backend.flux.util import configs as flux_configs from invokeai.backend.quantization.bnb_llm_int8 import quantize_model_llm_int8 from invokeai.backend.quantization.scripts.load_flux_model_bnb_nf4 import log_time