From 0b353a878a7a4bcea66da8a545189f956cc9bfe0 Mon Sep 17 00:00:00 2001
From: lvmin <lyuminzhang@outlook.com>
Date: Thu, 10 Aug 2023 05:32:22 -0700
Subject: [PATCH] i

---
 webui.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/webui.py b/webui.py
index 682c5218..4a2a0a11 100644
--- a/webui.py
+++ b/webui.py
@@ -22,6 +22,7 @@ del xl_base_clipvision
 opCLIPTextEncode = CLIPTextEncode()
 opEmptyLatentImage = EmptyLatentImage()
 opKSamplerAdvanced = KSamplerAdvanced()
+opVAEDecode = VAEDecode()
 
 positive_conditions = opCLIPTextEncode.encode(clip=xl_base_clip, text='a handsome man in forest')[0]
 negative_conditions = opCLIPTextEncode.encode(clip=xl_base_clip, text='bad, ugly')[0]
@@ -42,6 +43,8 @@ samples = opKSamplerAdvanced.sample(
     positive=positive_conditions,
     negative=negative_conditions,
     latent_image=initial_latent_image,
-)
+)[0]
+
+vaedecodeed = opVAEDecode.decode(samples=samples, vae=xl_base_vae)[0]
 
 a = 0