From 48034a3072bc4ddcdd2c7eb4450087d22bf50391 Mon Sep 17 00:00:00 2001
From: lvmin <lyuminzhang@outlook.com>
Date: Thu, 10 Aug 2023 05:50:36 -0700
Subject: [PATCH] i

---
 webui.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/webui.py b/webui.py
index 19e1a403..7e7cdce3 100644
--- a/webui.py
+++ b/webui.py
@@ -1,6 +1,7 @@
 import os
 import random
 import torch
+import numpy as np
 
 from comfy.sd import load_checkpoint_guess_config
 from nodes import VAEDecode, KSamplerAdvanced, EmptyLatentImage, SaveImage, CLIPTextEncode
@@ -42,4 +43,8 @@ with torch.no_grad():
 
     vae_decoded = opVAEDecode.decode(samples=samples, vae=xl_base_vae)[0]
 
-    a = 0
+    for image in vae_decoded:
+        i = 255. * image.cpu().numpy()
+        img = np.clip(i, 0, 255).astype(np.uint8)
+        import cv2
+        cv2.imwrite('a.png', img[:, :, ::-1])