fix: xformers

modules/import_hook.py

@@ -0,0 +1,18 @@
+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

webui.py

@@ -8,6 +8,7 @@ from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.middleware.gzip import GZipMiddleware
 
+from modules import import_hook
 from modules.call_queue import wrap_queued_call, queue_lock, wrap_gradio_gpu_call
 from modules.paths import script_path