update sd_hijack_opt to respect new env variables

This commit is contained in:
C43H66N12O12S2
2022-10-08 16:21:40 +03:00
committed by GitHub
parent ddfa9a9786
commit 69d0053583

View File

@@ -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