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

Intergrate Native AutoEncoderKL

Browse Source
This commit is contained in:
layerdiffusion
2024-07-31 21:08:22 -07:00
parent 2c3afff371
commit 0d079a846d
9 changed files with 473 additions and 93 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules_forge/forge_loader.py
View File

@@ -109,7 +109,7 @@ def load_checkpoint_guess_config(sd, output_vae=True, output_clip=True, output_c
if output_vae:
vae = huggingface_components['vae']
vae = VAE(model=vae, mapping=vae.state_dict_mapping)
vae = VAE(model=vae)
if output_clip:
w = WeightsLoader()