diff options
| author | C43H66N12O12S2 <36072735+C43H66N12O12S2@users.noreply.github.com> | 2022-10-08 13:21:40 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-08 13:21:40 +0000 |
| commit | 69d0053583757ce2942d62de81e8b89e6be07840 (patch) | |
| tree | 4392725e0a025380471d182db0599458b83c78b5 /modules/sd_hijack_optimizations.py | |
| parent | ddfa9a97865c732193023a71521c5b7b53d8571b (diff) | |
| download | stable-diffusion-webui-gfx803-69d0053583757ce2942d62de81e8b89e6be07840.tar.gz stable-diffusion-webui-gfx803-69d0053583757ce2942d62de81e8b89e6be07840.tar.bz2 stable-diffusion-webui-gfx803-69d0053583757ce2942d62de81e8b89e6be07840.zip | |
update sd_hijack_opt to respect new env variables
Diffstat (limited to 'modules/sd_hijack_optimizations.py')
| -rw-r--r-- | modules/sd_hijack_optimizations.py | 11 |
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
|