| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687 |
- import sys
- import gradio as gr
- from modules import shared_cmd_options, shared_gradio_themes, options, shared_items
- 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 ldm.models.diffusion.ddpm import LatentDiffusion
- from modules import util
- cmd_opts = shared_cmd_options.cmd_opts
- parser = shared_cmd_options.parser
- batch_cond_uncond = cmd_opts.always_batch_cond_uncond or not (cmd_opts.lowvram or cmd_opts.medvram)
- parallel_processing_allowed = not cmd_opts.lowvram and not cmd_opts.medvram
- 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
- sd_model: LatentDiffusion = None
- settings_components = None
- """assinged from ui.py, a mapping on setting names to gradio components repsponsible 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 = None
- 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
|
