Merge pull request #4488 from d8ahazard/ExtensionPreload

import os

import sys

import traceback

from importlib.machinery import SourceFileLoader

import git

from modules import paths, shared

extensions = []

extensions_dir = os.path.join(paths.script_path, "extensions")

        extension = Extension(name=dirname, path=path, enabled=dirname not in shared.opts.disabled_extensions)

        extensions.append(extension)

def preload_extensions(parser):

    if not os.path.isdir(extensions_dir):

        return

    for dirname in sorted(os.listdir(extensions_dir)):

        path = os.path.join(extensions_dir, dirname)

        if not os.path.isdir(path):

            continue

        for file in os.listdir(path):

            if "preload.py" in file:

                full_file = os.path.join(path, file)

                print(f"Got preload file: {full_file}")

                try:

                    ext = SourceFileLoader("preload", full_file).load_module()

                    parser = ext.preload(parser)

                except Exception as e:

                    print(f"Exception preloading script: {e}")

    return parser

 No newline at end of file

import modules.sd_models

import modules.styles

import modules.devices as devices

from modules import sd_samplers, sd_models, localization, sd_vae

from modules import sd_samplers, sd_models, localization, sd_vae, extensions

from modules.hypernetworks import hypernetwork

from modules.paths import models_path, script_path, sd_path

parser.add_argument("--tls-certfile", type=str, help="Partially enables TLS, requires --tls-keyfile to fully function", default=None)

parser.add_argument("--server-name", type=str, help="Sets hostname of server", default=None)

extensions.preload_extensions(parser)

cmd_opts = parser.parse_args()

restricted_opts = {

    "samples_filename_pattern",

    "directories_filename_pattern",