Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 00:34:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

error handle vae decode

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-08-24 10:26:28 +02:00
parent 8eefb62173
commit fb5584d3fe
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_vae.py
+4 -1
View File
@@ -141,7 +141,10 @@ def full_vae_decode(latents, model):
log_debug(f'VAE config: {model.vae.config}')
try:
with devices.inference_context():
decoded = model.vae.decode(latents, return_dict=False)[0]
decoded = model.vae.decode(latents, return_dict=False)
if decoded is None or len(decoded) == 0:
raise RuntimeError('returned no data')
decoded = decoded[0]
except Exception as e:
log.error(f'VAE decode: {e}')
if 'out of memory' not in str(e) and 'no data' not in str(e):