lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-01-26 19:09:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

improve vae key mapping

Browse Source
This commit is contained in:
layerdiffusion
2024-07-30 09:23:58 -06:00
parent 3289ccb53f
commit 40dd61ba6c
4 changed files with 23 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules_forge/forge_loader.py
View File

@@ -106,8 +106,8 @@ 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, mapping = load_vae_from_state_dict(sd)
vae = VAE(model=vae, mapping=mapping)
vae = load_vae_from_state_dict(sd)
vae = VAE(model=vae, mapping=vae.state_dict_mapping)
if output_clip:
w = WeightsLoader()