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
Download Patch File Download Diff File Expand all files Collapse all files

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