Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

redesign logging

Browse Source
This commit is contained in:
Vladimir Mandic
2023-05-02 13:57:16 -04:00
parent 2166b4de06
commit cb4cff3929
31 changed files with 236 additions and 354 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/import_hook.py
+4 -4
View File
@@ -1,15 +1,15 @@
import sys
from modules.shared import opts
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 xformers.ops # pylint: disable=unused-import
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:
print('Unloading xFormers')
log.info('Unloading xFormers')
sys.modules["xformers"] = None
sys.modules["xformers.ops"] = None