github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

Run Ruff

Browse Source
This commit is contained in:
Brandon Rising
2024-08-15 10:27:42 -04:00
committed by Brandon
parent 9ed53af520
commit 2d9042fb93
4 changed files with 5 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/quantization/fast_quantized_diffusion_model.py
View File

@@ -3,6 +3,7 @@ import os
from typing import Union
from diffusers.models.model_loading_utils import load_state_dict
from diffusers.models.transformers.transformer_flux import FluxTransformer2DModel
from diffusers.utils import (
CONFIG_NAME,
SAFE_WEIGHTS_INDEX_NAME,
@@ -12,7 +13,6 @@ from diffusers.utils import (
)
from optimum.quanto.models import QuantizedDiffusersModel
from optimum.quanto.models.shared_dict import ShardedStateDict
from diffusers.models.transformers.transformer_flux import FluxTransformer2DModel
from invokeai.backend.requantize import requantize