Fix model archs with no VAE

This commit is contained in:
Disty0
2024-02-13 21:48:11 +03:00
parent 08567b288f
commit 5fe82c422f
2 changed files with 4 additions and 2 deletions
+3 -1
View File
@@ -595,8 +595,10 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
if not hasattr(output, 'images') and hasattr(output, 'frames'):
shared.log.debug(f'Generated: frames={len(output.frames[0])}')
output.images = output.frames[0]
if output.images is not None and len(output.images) > 0:
if hasattr(shared.sd_model, "vae") and output.images is not None and len(output.images) > 0:
results = processing_vae.vae_decode(latents=output.images, model=shared.sd_model, full_quality=p.full_quality)
elif hasattr(output, 'images'):
results = output.images
else:
shared.log.warning('Processing returned no results')
results = []
@@ -184,7 +184,7 @@ class EmbeddingDatabase:
if ext.upper() not in exts:
raise ValueError(f'extension `{ext}` is invalid, expected one of: {exts}')
if name in tokenizer_vocab:
raise ValueError(f'invalid embedding name (cannot exist in vocab.json)')
raise ValueError('invalid embedding name (cannot exist in vocab.json)')
embeddings_to_load.append(embedding)
except Exception as e:
skipped_embeddings.append(embedding)