Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix incorrect VAE config file path during conversion of ckpts

Browse Source
This commit is contained in:
Lincoln Stein 2023-06-30 19:14:06 -04:00
parent 8a90e51408
commit 1347fc2f00
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/backend/model_management/models/vae.py
View File

@ -137,7 +137,6 @@ def _convert_vae_ckpt_and_cache(
from .stable_diffusion import _select_ckpt_config from .stable_diffusion import _select_ckpt_config
# all sd models use same vae settings # all sd models use same vae settings
config_file = _select_ckpt_config(base_model, ModelVariantType.Normal) config_file = _select_ckpt_config(base_model, ModelVariantType.Normal)
else: else:
raise Exception(f"Vae conversion not supported for model type: {base_model}") raise Exception(f"Vae conversion not supported for model type: {base_model}")
@ -152,7 +151,7 @@ def _convert_vae_ckpt_and_cache(
if "state_dict" in checkpoint: if "state_dict" in checkpoint:
checkpoint = checkpoint["state_dict"] checkpoint = checkpoint["state_dict"]
config = OmegaConf.load(config_file) config = OmegaConf.load(app_config.root_path/config_file)
vae_model = convert_ldm_vae_to_diffusers( vae_model = convert_ldm_vae_to_diffusers(
checkpoint = checkpoint, checkpoint = checkpoint,