update sd_hijack_opt to respect new env variables

1 files changed, 8 insertions, 3 deletions

diff --git a/modules/sd_hijack_optimizations.py b/modules/sd_hijack_optimizations.py
index ee58c7e4..be09ec8f 100644
--- a/modules/sd_hijack_optimizations.py
+++ b/modules/sd_hijack_optimizations.py
@@ -1,9 +1,14 @@
 import math

 import torch

 from torch import einsum

-import xformers.ops

-import functorch

-xformers._is_functorch_available=True

+try:

+    import xformers.ops

+    import functorch

+    xformers._is_functorch_available = True

+    shared.xformers_available = True

+except:

+    print('Cannot find xformers, defaulting to split attention. Try setting --xformers in your webui-user file if you wish to install it.')

+    continue

 from ldm.util import default

 from einops import rearrange