stable-diffusion-webui/modules/shared.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

88 lines
2.4 KiB
Python
Raw Normal View History

import sys
import gradio as gr
from modules import shared_cmd_options, shared_gradio_themes, options, shared_items, sd_models_types
2023-05-10 00:02:23 -06:00
from modules.paths_internal import models_path, script_path, data_path, sd_configs_path, sd_default_config, sd_model_file, default_sd_model_file, extensions_dir, extensions_builtin_dir # noqa: F401
from modules import util
cmd_opts = shared_cmd_options.cmd_opts
parser = shared_cmd_options.parser
2022-09-10 23:11:27 -06:00
batch_cond_uncond = True # old field, unused now in favor of shared.opts.batch_cond_uncond
2023-08-22 09:49:08 -06:00
parallel_processing_allowed = True
styles_filename = cmd_opts.styles_file
config_filename = cmd_opts.ui_settings_file
hide_dirs = {"visible": not cmd_opts.hide_ui_dir_config}
demo = None
device = None
weight_load_location = None
xformers_available = False
hypernetworks = {}
loaded_hypernetworks = []
state = None
prompt_styles = None
interrogator = None
face_restorers = []
options_templates = None
opts = None
restricted_opts = None
sd_model: sd_models_types.WebuiSdModel = None
settings_components = None
"""assinged from ui.py, a mapping on setting names to gradio components repsponsible for those settings"""
tab_names = []
2023-01-02 09:42:10 -07:00
latent_upscale_default_mode = "Latent"
latent_upscale_modes = {
2023-01-04 03:12:06 -07:00
"Latent": {"mode": "bilinear", "antialias": False},
"Latent (antialiased)": {"mode": "bilinear", "antialias": True},
"Latent (bicubic)": {"mode": "bicubic", "antialias": False},
"Latent (bicubic antialiased)": {"mode": "bicubic", "antialias": True},
2023-01-04 03:12:06 -07:00
"Latent (nearest)": {"mode": "nearest", "antialias": False},
"Latent (nearest-exact)": {"mode": "nearest-exact", "antialias": False},
2023-01-02 09:42:10 -07:00
}
2022-09-04 09:54:12 -06:00
sd_upscalers = []
2022-10-16 09:53:56 -06:00
clip_model = None
2022-09-08 07:37:13 -06:00
progress_print_out = sys.stdout
2023-03-25 14:11:41 -06:00
gradio_theme = gr.themes.Base()
total_tqdm = None
2023-03-25 14:11:41 -06:00
mem_mon = 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
ldm_print = util.ldm_print
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
reload_hypernetworks = shared_items.reload_hypernetworks