From 1c70056744b4fe1a88ec443e1a94b59b8fcc93b0 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Thu, 18 May 2023 19:09:34 -0400 Subject: [PATCH] match vae file --- modules/sd_vae.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/modules/sd_vae.py b/modules/sd_vae.py index 7bd294c6d..0e8db42ea 100644 --- a/modules/sd_vae.py +++ b/modules/sd_vae.py @@ -96,7 +96,14 @@ def resolve_vae(checkpoint_file): if vae_near_checkpoint is not None and (shared.opts.sd_vae_as_default): return vae_near_checkpoint, 'near checkpoint' if is_automatic: - for named_vae_location in [os.path.join(vae_path, os.path.splitext(os.path.basename(checkpoint_file))[0] + ".vae.pt"), os.path.join(vae_path, os.path.splitext(os.path.basename(checkpoint_file))[0] + ".vae.ckpt"), os.path.join(vae_path, os.path.splitext(os.path.basename(checkpoint_file))[0] + ".vae.safetensors")]: + for named_vae_location in [ + os.path.join(vae_path, os.path.splitext(os.path.basename(checkpoint_file))[0] + ".pt"), + os.path.join(vae_path, os.path.splitext(os.path.basename(checkpoint_file))[0] + ".ckpt"), + os.path.join(vae_path, os.path.splitext(os.path.basename(checkpoint_file))[0] + ".safetensors"), + os.path.join(vae_path, os.path.splitext(os.path.basename(checkpoint_file))[0] + ".vae.pt"), + os.path.join(vae_path, os.path.splitext(os.path.basename(checkpoint_file))[0] + ".vae.ckpt"), + os.path.join(vae_path, os.path.splitext(os.path.basename(checkpoint_file))[0] + ".vae.safetensors"), + ]: if os.path.isfile(named_vae_location): return named_vae_location, 'in VAE dir' if shared.opts.sd_vae == "None":