lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-23 00:03:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

rework model loader and configs

Browse Source
This commit is contained in:
layerdiffusion
2024-07-30 09:57:07 -06:00
parent 40dd61ba6c
commit c8156fcf41
181 changed files with 1853633 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modules_forge/forge_loader.py
View File

@@ -18,7 +18,7 @@ from ldm.util import instantiate_from_config
from modules_forge import forge_clip
from modules_forge.unet_patcher import UnetPatcher
from ldm_patched.modules.model_base import model_sampling, ModelType
from backend.vae.loader import load_vae_from_state_dict
from backend.loader import load_huggingface_components
import open_clip
from transformers import CLIPTextModel, CLIPTokenizer
@@ -99,6 +99,8 @@ def load_checkpoint_guess_config(sd, output_vae=True, output_clip=True, output_c
if output_clipvision:
clipvision = ldm_patched.modules.clip_vision.load_clipvision_from_sd(sd, model_config.clip_vision_prefix, True)
huggingface_components = load_huggingface_components(sd)
if output_model:
inital_load_device = model_management.unet_inital_load_device(parameters, unet_dtype)
offload_device = model_management.unet_offload_device()
@@ -106,7 +108,7 @@ def load_checkpoint_guess_config(sd, output_vae=True, output_clip=True, output_c
model.load_model_weights(sd, "model.diffusion_model.")
if output_vae:
vae = load_vae_from_state_dict(sd)
vae = huggingface_components['vae']
vae = VAE(model=vae, mapping=vae.state_dict_mapping)
if output_clip: