mrq/stable-diffusion-webui
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: xformers use importlib

Browse Source
This commit is contained in:
Akiba 2022-12-24 11:17:21 +08:00
parent 35e1017e3e
commit 13e0295ab6
No known key found for this signature in database
GPG Key ID: 9D600258808ACBCD
1 changed files with 1 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
modules/import_hook.py
View File

@ -1,18 +1,5 @@
import builtins
import sys import sys
old_import = builtins.__import__
IMPORT_BLACKLIST = []
if "xformers" not in "".join(sys.argv): if "xformers" not in "".join(sys.argv):
IMPORT_BLACKLIST.append("xformers") sys.modules["xformers"] = None
def import_hook(*args, **kwargs):
if args[0] in IMPORT_BLACKLIST:
raise ImportError("Import of %s is blacklisted" % args[0])
return old_import(*args, **kwargs)
builtins.__import__ = import_hook