update sd_hijack_opt to respect new env variables
This commit is contained in:
parent
ddfa9a9786
commit
69d0053583
1 changed files with 8 additions and 3 deletions
|
@ -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
|
||||
|
||||
|
|
Loading…
Reference in a new issue