StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update sd_hijack_opt to respect new env variables

Browse Source
This commit is contained in:
C43H66N12O12S2 2022-10-08 16:21:40 +03:00 committed by GitHub
parent ddfa9a9786
commit 69d0053583
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
modules/sd_hijack_optimizations.py
View File

@ -1,9 +1,14 @@
import math import math
import torch import torch
from torch import einsum from torch import einsum
import xformers.ops try:
import functorch import xformers.ops
xformers._is_functorch_available=True 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 ldm.util import default
from einops import rearrange from einops import rearrange