diff options
| author | catboxanon <122327233+catboxanon@users.noreply.github.com> | 2023-08-10 16:04:49 -0400 |
|---|---|---|
| committer | catboxanon <122327233+catboxanon@users.noreply.github.com> | 2023-08-10 16:04:49 -0400 |
| commit | d456fb797ad9e1f6daddbdaf284ae34cfb2a0656 (patch) | |
| tree | 3a8bd56d9ae7a0bfafe1ac7ea825b9b9d077333e | |
| parent | 458eda13211ac3498485f1e5154d90808fbcfb60 (diff) | |
fix: Properly return None when VAE hash is None
| -rw-r--r-- | modules/sd_vae.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/modules/sd_vae.py b/modules/sd_vae.py index 1db01992..fd9a1c2a 100644 --- a/modules/sd_vae.py +++ b/modules/sd_vae.py @@ -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): |