Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

ruff

Browse Source
This commit is contained in:
Ryan Dick 2024-08-26 14:09:02 +00:00 committed by Brandon
parent 70c278c810
commit 635d2f480d
3 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/app/invocations/model.py
View File

@ -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,

2
invokeai/backend/model_manager/load/model_loaders/flux.py
View File

@ -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,

4
invokeai/backend/quantization/scripts/load_flux_model_bnb_llm_int8.py
View File

@ -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