diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-22 19:05:03 +0300 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-22 19:05:03 +0300 |
| commit | 3ec5ce941626c3c5ff4aac884c926ceb4ba60d37 (patch) | |
| tree | 3d44d84afa9cf7efec71c8ecf9bd247fc3801a44 /modules/shared.py | |
| parent | 016554e43740e0b7ded75e89255de81270de9d6c (diff) | |
add type annotations for extra fields of shared.sd_model
Diffstat (limited to 'modules/shared.py')
| -rw-r--r-- | modules/shared.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/modules/shared.py b/modules/shared.py index f321159d..63661939 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -2,9 +2,8 @@ import sys import gradio as gr
-from modules import shared_cmd_options, shared_gradio_themes, options, shared_items
+from modules import shared_cmd_options, shared_gradio_themes, options, shared_items, sd_models_types
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
@@ -40,7 +39,7 @@ options_templates = None opts = None
restricted_opts = None
-sd_model: LatentDiffusion = 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"""
|