Unverified Commit 35e1017e authored by Akiba's avatar Akiba
Browse files

fix: xformers

parent 685f9631
Loading
Loading
Loading
Loading

modules/import_hook.py

0 → 100644
+18 −0
Original line number Diff line number Diff line
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
+1 −0
Original line number Diff line number Diff line
@@ -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