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 import copy
from typing import List, Literal, Optional from typing import List, Literal, Optional
import yaml
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from invokeai.app.invocations.baseinvocation import ( from invokeai.app.invocations.baseinvocation import (
@ -11,10 +10,10 @@ from invokeai.app.invocations.baseinvocation import (
invocation, invocation,
invocation_output, 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.invocations.fields import FieldDescriptions, Input, InputField, OutputField, UIType
from invokeai.app.services.shared.invocation_context import InvocationContext from invokeai.app.services.shared.invocation_context import InvocationContext
from invokeai.app.shared.models import FreeUConfig from invokeai.app.shared.models import FreeUConfig
from invokeai.backend.flux.util import max_seq_lengths
from invokeai.backend.model_manager.config import ( from invokeai.backend.model_manager.config import (
AnyModelConfig, AnyModelConfig,
BaseModelType, 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.app.services.config.config_default import get_config
from invokeai.backend.flux.model import Flux 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.modules.autoencoder import AutoEncoder
from invokeai.backend.flux.util import ae_params, configs
from invokeai.backend.model_manager import ( from invokeai.backend.model_manager import (
AnyModel, AnyModel,
AnyModelConfig, AnyModelConfig,

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

@ -1,10 +1,10 @@
from pathlib import Path from pathlib import Path
import accelerate 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 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.bnb_llm_int8 import quantize_model_llm_int8
from invokeai.backend.quantization.scripts.load_flux_model_bnb_nf4 import log_time from invokeai.backend.quantization.scripts.load_flux_model_bnb_nf4 import log_time