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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 3 deletions

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