add type annotations for extra fields of shared.sd_model

This commit is contained in:
AUTOMATIC1111
2023-08-22 19:05:03 +03:00
parent 016554e437
commit 3ec5ce9416
2 changed files with 33 additions and 3 deletions

View File

@@ -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"""