mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-01-30 12:22:26 -08:00
Code Issues Projects Releases Packages Wiki Activity

Update sd_vae.py

Browse source 
There is no need to use split.
This commit is contained in:
Sakura-Luna 2023-05-11 14:29:52 +08:00 • committed by GitHub
parent 749a93295e
commit 1dcd672324
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
modules/sd_vae.py
View file

@ -88,10 +88,9 @@ def refresh_vae_list():
def find_vae_near_checkpoint(checkpoint_file):
checkpoint_path = os.path.basename(checkpoint_file).split('.', 1)[0]
checkpoint_path = os.path.basename(checkpoint_file).rsplit('.', 1)[0]
for vae_file in vae_dict.values():
vae_path = os.path.basename(vae_file).split('.', 1)[0]
if vae_path == checkpoint_path:
if os.path.basename(vae_file).startswith(checkpoint_path):
return vae_file
return None