You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
19 lines
379 B
Python
19 lines
379 B
Python
import builtins
|
|
import sys
|
|
|
|
old_import = builtins.__import__
|
|
IMPORT_BLACKLIST = []
|
|
|
|
|
|
if "xformers" not in "".join(sys.argv):
|
|
IMPORT_BLACKLIST.append("xformers")
|
|
|
|
|
|
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
|