Skip to content

Commit

Permalink
type hintinh in shared.py
Browse files Browse the repository at this point in the history
  • Loading branch information
light-and-ray committed Mar 11, 2024
1 parent 3e0146f commit 7c89bd1
Showing 1 changed file with 16 additions and 12 deletions.
28 changes: 16 additions & 12 deletions modules/shared.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,10 @@
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

falseVar = False # avoid circular import for type hiting
if falseVar:
from modules import shared_state, styles, interrogate, shared_total_tqdm, memmon

cmd_opts = shared_cmd_options.cmd_opts
parser = shared_cmd_options.parser

Expand All @@ -16,33 +20,33 @@
config_filename = cmd_opts.ui_settings_file
hide_dirs = {"visible": not cmd_opts.hide_ui_dir_config}

demo = None
demo: gr.Blocks = None

device = None
device: str = None

weight_load_location = None
weight_load_location: str = None

xformers_available = False

hypernetworks = {}

loaded_hypernetworks = []

state = None
state: 'shared_state.State' = None

prompt_styles = None
prompt_styles: 'styles.StyleDatabase' = None

interrogator = None
interrogator: 'interrogate.InterrogateModels' = None

face_restorers = []

options_templates = None
opts = None
restricted_opts = None
options_templates: dict = None
opts: options.Options = None
restricted_opts: set[str] = None

sd_model: sd_models_types.WebuiSdModel = None

settings_components = None
settings_components: dict = None
"""assigned from ui.py, a mapping on setting names to gradio components repsponsible for those settings"""

tab_names = []
Expand All @@ -65,9 +69,9 @@

gradio_theme = gr.themes.Base()

total_tqdm = None
total_tqdm: 'shared_total_tqdm.TotalTQDM' = None

mem_mon = None
mem_mon: 'memmon.MemUsageMonitor' = None

options_section = options.options_section
OptionInfo = options.OptionInfo
Expand Down

0 comments on commit 7c89bd1

Please sign in to comment.