add type annotations for extra fields of shared.sd_model

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

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

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