Commit 69d00535 authored by C43H66N12O12S2's avatar C43H66N12O12S2 Committed by GitHub

update sd_hijack_opt to respect new env variables

parent ddfa9a97
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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment