stable-diffusion-webui/modules/shared.py

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

95 lines
2.9 KiB
Python
Raw Normal View History

import os
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
from typing import TYPE_CHECKING
if TYPE_CHECKING:
2024-03-10 19:15:09 -06:00
from modules import shared_state, styles, interrogate, shared_total_tqdm, memmon
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 = cmd_opts.styles_file if len(cmd_opts.styles_file) > 0 else [os.path.join(data_path, 'styles.csv')]
config_filename = cmd_opts.ui_settings_file
hide_dirs = {"visible": not cmd_opts.hide_ui_dir_config}
2024-03-10 19:15:09 -06:00
demo: gr.Blocks = None
2024-03-10 19:15:09 -06:00
device: str = None
2024-03-10 19:15:09 -06:00
weight_load_location: str = None
xformers_available = False
hypernetworks = {}
loaded_hypernetworks = []
2024-03-10 19:15:09 -06:00
state: 'shared_state.State' = None
2024-03-10 19:15:09 -06:00
prompt_styles: 'styles.StyleDatabase' = None
2024-03-10 19:15:09 -06:00
interrogator: 'interrogate.InterrogateModels' = None
face_restorers = []
2024-03-10 19:15:09 -06:00
options_templates: dict = None
opts: options.Options = None
restricted_opts: set[str] = None
sd_model: sd_models_types.WebuiSdModel = None
2024-03-10 19:15:09 -06:00
settings_components: dict = None
2024-03-03 23:37:23 -07:00
"""assigned 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()
2024-03-10 19:15:09 -06:00
total_tqdm: 'shared_total_tqdm.TotalTQDM' = None
2023-03-25 14:11:41 -06:00
2024-03-10 19:15:09 -06:00
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
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
2024-04-05 23:53:21 -06:00
hf_endpoint = os.getenv('HF_ENDPOINT', 'https://huggingface.co')