mirror of
https://github.com/openvinotoolkit/stable-diffusion-webui.git
synced 2024-12-14 22:53:25 +03:00
add type annotations for extra fields of shared.sd_model
This commit is contained in:
parent
016554e437
commit
3ec5ce9416
31
modules/sd_models_types.py
Normal file
31
modules/sd_models_types.py
Normal file
@ -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"""
|
@ -2,9 +2,8 @@ import sys
|
|||||||
|
|
||||||
import gradio as gr
|
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 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
|
from modules import util
|
||||||
|
|
||||||
cmd_opts = shared_cmd_options.cmd_opts
|
cmd_opts = shared_cmd_options.cmd_opts
|
||||||
@ -40,7 +39,7 @@ options_templates = None
|
|||||||
opts = None
|
opts = None
|
||||||
restricted_opts = None
|
restricted_opts = None
|
||||||
|
|
||||||
sd_model: LatentDiffusion = None
|
sd_model: sd_models_types.WebuiSdModel = None
|
||||||
|
|
||||||
settings_components = None
|
settings_components = None
|
||||||
"""assinged from ui.py, a mapping on setting names to gradio components repsponsible for those settings"""
|
"""assinged from ui.py, a mapping on setting names to gradio components repsponsible for those settings"""
|
||||||
|
Loading…
Reference in New Issue
Block a user