VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Wrong indention

Browse Source
This commit is contained in:
Hacker 17082006
2023-01-14 14:59:29 +07:00
parent 95ee349e2a
commit 4fe1dd6a1c
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
library/model_util.py
View File

@@ -1106,12 +1106,12 @@ def load_vae(vae_id, dtype):
if vae_key.startswith(VAE_PREFIX): if vae_key.startswith(VAE_PREFIX):
full_model = True full_model = True
break break
if not full_model: if not full_model:
sd = {} sd = {}
for key, value in vae_sd.items(): for key, value in vae_sd.items():
sd[VAE_PREFIX + key] = value sd[VAE_PREFIX + key] = value
vae_sd = sd vae_sd = sd
del sd del sd
converted_vae_checkpoint = convert_ldm_vae_checkpoint(vae_sd, vae_config) converted_vae_checkpoint = convert_ldm_vae_checkpoint(vae_sd, vae_config)
vae = AutoencoderKL(**vae_config) vae = AutoencoderKL(**vae_config)