Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

i

Browse Source
This commit is contained in:
lvmin
2023-08-10 05:50:36 -07:00
parent fbd3a3ed50
commit 48034a3072
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
webui.py
+6 -1
View File
@@ -1,6 +1,7 @@
import os import os
import random import random
import torch import torch
import numpy as np
from comfy.sd import load_checkpoint_guess_config from comfy.sd import load_checkpoint_guess_config
from nodes import VAEDecode, KSamplerAdvanced, EmptyLatentImage, SaveImage, CLIPTextEncode 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] 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])