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

fix alphavllm-lumina

Browse Source
This commit is contained in:
Vladimir Mandic
2024-09-21 11:56:11 -04:00
parent 0b665fe4ae
commit 8457d45e26
4 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_vae.py
+9 -5
View File
@@ -1,5 +1,6 @@
import os
import time
import numpy as np
import torch
import torchvision.transforms.functional as TF
from modules import shared, devices, sd_models, sd_vae, sd_vae_taesd
@@ -157,12 +158,15 @@ def vae_decode(latents, model, output_type='np', full_quality=True, width=None,
else:
decoded = taesd_vae_decode(latents=latents)
if hasattr(model, 'image_processor'):
imgs = model.image_processor.postprocess(decoded, output_type=output_type)
if torch.is_tensor(decoded):
if hasattr(model, 'image_processor'):
imgs = model.image_processor.postprocess(decoded, output_type=output_type)
else:
import diffusers
model.image_processor = diffusers.image_processor.VaeImageProcessor()
imgs = model.image_processor.postprocess(decoded, output_type=output_type)
else:
import diffusers
model.image_processor = diffusers.image_processor.VaeImageProcessor()
imgs = model.image_processor.postprocess(decoded, output_type=output_type)
imgs = decoded if isinstance(decoded, list) or isinstance(decoded, np.ndarray) else [decoded]
shared.state.job = prev_job
if shared.cmd_opts.profile or debug: