hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 03:10:21 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

files in vae folder with same name as a checkpoint can be found too

Browse Source
This commit is contained in:
Micky Brunetti
2023-05-09 15:35:57 +02:00
committed by GitHub
parent 5ab7f213be
commit 7fd3a4e6d7
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
modules/sd_vae.py
View File

@@ -88,10 +88,13 @@ def refresh_vae_list():
def find_vae_near_checkpoint(checkpoint_file):
checkpoint_path = os.path.splitext(checkpoint_file)[0]
for vae_location in [checkpoint_path + ".vae.pt", checkpoint_path + ".vae.ckpt", checkpoint_path + ".vae.safetensors"]:
if os.path.isfile(vae_location):
return vae_location
checkpoint_path = os.path.basename(checkpoint_file).split('.', 1)[0]
print(f"checkpoint: {checkpoint_path}")
for vae_file in vae_dict.values():
vae_path = os.path.basename(vae_file).split('.', 1)[0]
print(f"vae: {vae_path}")
if vae_path == checkpoint_path:
return vae_file
return None