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

Blackify

Browse Source
This commit is contained in:
Sergey Borisov 2023-08-28 18:36:27 +03:00
parent cd548f73fd
commit 2bf747caf6
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/util/hotfixes.py
View File

@ -764,6 +764,7 @@ diffusers.models.controlnet.ControlNetModel = ControlNetModel
try: try:
import xformers import xformers
xformers_available = True xformers_available = True
except: except:
xformers_available = False xformers_available = False
@ -772,6 +773,7 @@ except:
if xformers_available: if xformers_available:
# TODO: remove when fixed in diffusers # TODO: remove when fixed in diffusers
_xformers_memory_efficient_attention = xformers.ops.memory_efficient_attention _xformers_memory_efficient_attention = xformers.ops.memory_efficient_attention
def new_memory_efficient_attention( def new_memory_efficient_attention(
query: torch.Tensor, query: torch.Tensor,
key: torch.Tensor, key: torch.Tensor,