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

Fix hopefully last import

Browse Source
This commit is contained in:
Wubbbi 2024-02-01 14:56:17 +01:00 committed by Kent Keirsey
parent 413fe566b8
commit 5e1e50bd47
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
invokeai/backend/stable_diffusion/diffusion/cross_attention_control.py
View File

@ -1,5 +1,4 @@
# adapted from bloc97's CrossAttentionControl colab
# https://github.com/bloc97/CrossAttentionControl
# adapted from bloc97's CrossAttentionControl colab# https://github.com/bloc97/CrossAttentionControl
import enum
@ -12,7 +11,7 @@ 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.unets.unet_2d_condition import UNet2DConditionModel
from torch import nn
import invokeai.backend.util.logging as logger