aboutsummaryrefslogtreecommitdiff
path: root/modules/sd_models_xl.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-07-13 09:30:33 +0300
committerAUTOMATIC1111 <16777216c@gmail.com>2023-07-13 09:30:33 +0300
commitac4ccfa1369e74492b467294eab96c3f558b297b (patch)
tree357cc16bc3a82c2c4a07c410e3cd4e83e5e3b6f9 /modules/sd_models_xl.py
parentb717eb7e56a4e620e77a2225e80223c89cb4f0d1 (diff)
get attention optimizations to work
Diffstat (limited to 'modules/sd_models_xl.py')
-rw-r--r--modules/sd_models_xl.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/sd_models_xl.py b/modules/sd_models_xl.py
index 9224c1a3..4d1aa497 100644
--- a/modules/sd_models_xl.py
+++ b/modules/sd_models_xl.py
@@ -55,3 +55,6 @@ sgm.modules.diffusionmodules.model.print = lambda *args: None
sgm.modules.diffusionmodules.openaimodel.print = lambda *args: None
sgm.modules.encoders.modules.print = lambda *args: None
+# this gets the code to load the vanilla attention that we override
+sgm.modules.attention.SDP_IS_AVAILABLE = True
+sgm.modules.attention.XFORMERS_IS_AVAILABLE = False \ No newline at end of file