mirror of
https://github.com/vladmandic/automatic
synced 2026-08-26 23:20:59 +02:00
Vladimir Mandic
16 lines
542 B
Python
16 lines
542 B
Python
import sys
|
|
from modules.shared import opts, log
|
|
|
|
# this will break any attempt to import xformers which will prevent stability diffusion repo from trying to use it
|
|
try:
|
|
import xformers # pylint: disable=unused-import, import-error
|
|
import xformers.ops # pylint: disable=unused-import, import-error
|
|
except:
|
|
pass
|
|
|
|
if opts.cross_attention_optimization != "xFormers":
|
|
if sys.modules.get("xformers", None) is not None:
|
|
log.info('Unloading xFormers')
|
|
sys.modules["xformers"] = None
|
|
sys.modules["xformers.ops"] = None
|