mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-07-21 21:01:24 +08:00
85 lines
2.6 KiB
Python
85 lines
2.6 KiB
Python
import os
|
|
import sys
|
|
from typing import TYPE_CHECKING
|
|
|
|
import gradio as gr
|
|
|
|
from backend import memory_management
|
|
from modules import options, shared_cmd_options, shared_gradio_themes, shared_items, util
|
|
from modules.paths_internal import data_path, extensions_builtin_dir, extensions_dir, models_path, script_path # noqa: F401
|
|
|
|
if TYPE_CHECKING:
|
|
from modules import memmon, shared_state, shared_total_tqdm, styles
|
|
|
|
cmd_opts = shared_cmd_options.cmd_opts
|
|
parser = shared_cmd_options.parser
|
|
|
|
batch_cond_uncond = True # old field, unused now in favor of shared.opts.batch_cond_uncond
|
|
parallel_processing_allowed = True
|
|
styles_filename = cmd_opts.styles_file = cmd_opts.styles_file if len(cmd_opts.styles_file) > 0 else [os.path.join(data_path, "styles.csv"), os.path.join(data_path, "styles_integrated.csv")]
|
|
config_filename = cmd_opts.ui_settings_file
|
|
hide_dirs = {"visible": not cmd_opts.hide_ui_dir_config}
|
|
|
|
demo: gr.Blocks = None
|
|
|
|
device: str = None
|
|
|
|
xformers_available = memory_management.xformers_enabled()
|
|
|
|
state: "shared_state.State" = None
|
|
|
|
prompt_styles: "styles.StyleDatabase" = None
|
|
|
|
face_restorers = []
|
|
|
|
options_templates: dict = None
|
|
opts: options.Options = None
|
|
restricted_opts: set[str] = None
|
|
|
|
sd_model = None
|
|
|
|
settings_components: dict = None
|
|
"""assigned from ui.py, a mapping on setting names to gradio components responsible for those settings"""
|
|
|
|
tab_names = []
|
|
|
|
latent_upscale_default_mode = "Latent"
|
|
latent_upscale_modes = {
|
|
"Latent": {"mode": "bilinear", "antialias": False},
|
|
"Latent (antialiased)": {"mode": "bilinear", "antialias": True},
|
|
"Latent (bicubic)": {"mode": "bicubic", "antialias": False},
|
|
"Latent (bicubic antialiased)": {"mode": "bicubic", "antialias": True},
|
|
"Latent (nearest)": {"mode": "nearest", "antialias": False},
|
|
"Latent (nearest-exact)": {"mode": "nearest-exact", "antialias": False},
|
|
}
|
|
|
|
sd_upscalers = []
|
|
|
|
clip_model = None
|
|
|
|
progress_print_out = sys.stdout
|
|
|
|
gradio_theme = gr.themes.Base()
|
|
|
|
total_tqdm: "shared_total_tqdm.TotalTQDM" = None
|
|
|
|
mem_mon: "memmon.MemUsageMonitor" = None
|
|
|
|
options_section = options.options_section
|
|
OptionInfo = options.OptionInfo
|
|
OptionHTML = options.OptionHTML
|
|
|
|
natural_sort_key = util.natural_sort_key
|
|
listfiles = util.listfiles
|
|
html_path = util.html_path
|
|
html = util.html
|
|
walk_files = util.walk_files
|
|
|
|
reload_gradio_theme = shared_gradio_themes.reload_gradio_theme
|
|
|
|
list_checkpoint_tiles = shared_items.list_checkpoint_tiles
|
|
refresh_checkpoints = shared_items.refresh_checkpoints
|
|
list_samplers = shared_items.list_samplers
|
|
|
|
hf_endpoint = os.getenv("HF_ENDPOINT", "https://huggingface.co")
|