Mirrors
/
stable-diffusion-webui
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

py 3.9 find_vae()

This commit is contained in:
w-e-w 2024-07-08 14:07:26 +09:00
parent 1da4907927
commit 21e72d1a5e
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/xyz_grid.py
View File

@ -118,10 +118,9 @@ def apply_size(p, x: str, xs) -> None:
def find_vae(name: str): def find_vae(name: str):
match name := name.lower().strip(): if name := name.strip().lower() in ('auto', 'automatic'):
case 'auto', 'automatic':
return 'Automatic' return 'Automatic'
case 'none': elif name == 'none':
return 'None' return 'None'
return next((k for k in modules.sd_vae.vae_dict if k.lower() == name), print(f'No VAE found for {name}; using Automatic') or 'Automatic') return next((k for k in modules.sd_vae.vae_dict if k.lower() == name), print(f'No VAE found for {name}; using Automatic') or 'Automatic')