fix wildcards

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-12-22 08:20:16 -05:00
parent 042f5f9fed
commit dff846182f
3 changed files with 7 additions and 5 deletions
+2 -1
View File
@@ -144,7 +144,8 @@ def full_vae_decode(latents, model):
decoded = model.vae.decode(latents, return_dict=False)[0]
except Exception as e:
shared.log.error(f'VAE decode: {stats} {e}')
errors.display(e, 'VAE decode')
if 'out of memory' not in str(e):
errors.display(e, 'VAE decode')
decoded = []
if hasattr(model.vae, "orig_dtype"):