Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix a bug

Browse Source 
Forgot a return value which was causing 'ValueError: too many values to unpack (expected 2)' upon trying load a VAE with Automatic in sd_vae
This commit is contained in:
Gazzoo-byte
2023-04-25 21:00:03 +01:00
committed by GitHub
parent 70e4a1673a
commit 98c29302a7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_vae.py
+1 -1
View File
@@ -106,7 +106,7 @@ def resolve_vae(checkpoint_file):
for named_vae_location in [vae_path + "\\" + Path(checkpoint_file).stem + ".vae.pt", vae_path + "\\" + Path(checkpoint_file).stem + ".vae.ckpt", vae_path + "\\" + Path(checkpoint_file).stem + ".vae.safetensors"]:
if os.path.isfile(named_vae_location):
print(named_vae_location+' found in VAE dir')
return named_vae_location
return named_vae_location, ' found in VAE dir'
else:
print(f"Couldn't find a VAE with a matching name in {vae_path}, using None instead")
return None, None