isort wip 2

This commit is contained in:
Martin Kristiansen
2023-08-18 11:13:28 -04:00
committed by Kent Keirsey
parent 5615c31799
commit caea6d11c6
35 changed files with 168 additions and 180 deletions

View File

@ -4,8 +4,8 @@ Initialization file for invokeai.models.diffusion
from .cross_attention_control import InvokeAICrossAttentionMixin # noqa: F401
from .cross_attention_map_saving import AttentionMapSaver # noqa: F401
from .shared_invokeai_diffusion import ( # noqa: F401
BasicConditioningInfo,
InvokeAIDiffuserComponent,
PostprocessingSettings,
BasicConditioningInfo,
SDXLConditioningInfo,
)

View File

@ -11,16 +11,12 @@ import diffusers
import psutil
import torch
from compel.cross_attention_control import Arguments
from diffusers.models.attention_processor import Attention, AttentionProcessor, AttnProcessor, SlicedAttnProcessor
from diffusers.models.unet_2d_condition import UNet2DConditionModel
from diffusers.models.attention_processor import AttentionProcessor
from diffusers.models.attention_processor import (
Attention,
AttnProcessor,
SlicedAttnProcessor,
)
from torch import nn
import invokeai.backend.util.logging as logger
from ...util import torch_dtype

View File

@ -1,8 +1,8 @@
from __future__ import annotations
import math
from contextlib import contextmanager
from dataclasses import dataclass
import math
from typing import Any, Callable, Optional, Union
import torch