fix: Properly return None when VAE hash is None

This commit is contained in:
catboxanon 2023-08-10 16:04:49 -04:00
parent 458eda1321
commit d456fb797a
1 changed files with 3 additions and 1 deletions

View File

@ -31,7 +31,9 @@ def get_loaded_vae_hash():
if loaded_vae_file is None:
return None
return hashes.sha256(loaded_vae_file, 'vae')[0:10]
sha256 = hashes.sha256(loaded_vae_file, 'vae')
return sha256[0:10] if sha256 else None
def get_base_vae(model):