Unite attention optimization settings

This commit is contained in:
Disty0
2024-02-10 00:40:28 +03:00
committed by Vladimir Mandic
parent 71caa89246
commit 43c5be76ca
5 changed files with 351 additions and 207 deletions
+21 -10
View File
@@ -13,16 +13,27 @@ def refresh_vae_list():
modules.sd_vae.refresh_vae_list()
def list_crossattention():
return [
"Disabled",
"xFormers",
"Scaled-Dot-Product",
"Doggettx's",
"InvokeAI's",
"Sub-quadratic",
"Split attention"
]
def list_crossattention(diffusers=False):
if diffusers:
return [
"Disabled",
"xFormers",
"Scaled-Dot-Product",
"Torch BMM",
"Split attention",
"Dynamic Attention BMM",
"Dynamic Attention SDP"
]
else:
return [
"Disabled",
"xFormers",
"Scaled-Dot-Product",
"Doggettx's",
"InvokeAI's",
"Sub-quadratic",
"Split attention"
]
def get_pipelines():
import diffusers