add type annotations for extra fields of shared.sd_model

modules/sd_models_types.py

@@ -0,0 +1,31 @@
+from ldm.models.diffusion.ddpm import LatentDiffusion
+from typing import TYPE_CHECKING
+
+
+if TYPE_CHECKING:
+    from modules.sd_models import CheckpointInfo
+
+
+class WebuiSdModel(LatentDiffusion):
+    """This class is not actually instantinated, but its fields are created and fieeld by webui"""
+
+    lowvram: bool
+    """True if lowvram/medvram optimizations are enabled -- see modules.lowvram for more info"""
+
+    sd_model_hash: str
+    """short hash, 10 first characters of SHA1 hash of the model file; may be None if --no-hashing flag is used"""
+
+    sd_model_checkpoint: str
+    """path to the file on disk that model weights were obtained from"""
+
+    sd_checkpoint_info: 'CheckpointInfo'
+    """structure with additional information about the file with model's weights"""
+
+    is_sdxl: bool
+    """True if the model's architecture is SDXL"""
+
+    is_sd2: bool
+    """True if the model's architecture is SD 2.x"""
+
+    is_sd1: bool
+    """True if the model's architecture is SD 1.x"""

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