CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

typing

Browse Source
This commit is contained in:
Haoming 2026-05-28 11:03:19 +08:00
parent 4f643b5dc2
commit 1a03991313
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
backend/diffusion_engine/base.py
View File

@ -7,6 +7,7 @@ if TYPE_CHECKING:
from backend.patcher.clip import CLIP
from backend.patcher.unet import UnetPatcher
from backend.patcher.vae import VAE
from modules_forge.packages.huggingface_guess.model_list import BASE
from backend import memory_management, utils
@ -25,11 +26,11 @@ class ForgeObjects:
class ForgeDiffusionEngine:
matched_guesses = []
def __init__(self, estimated_config, huggingface_components):
def __init__(self, estimated_config: "BASE", huggingface_components: dict[str, "torch.nn.Module"]):
huggingface_components["vae"].latent_format = estimated_config.latent_format
self.model_config = estimated_config
self.is_inpaint = estimated_config.inpaint_model()
self.model_config: "BASE" = estimated_config
self.is_inpaint: bool = estimated_config.inpaint_model()
self.forge_objects: "ForgeObjects" = None
self.forge_objects_original: "ForgeObjects" = None