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 12:35:43 -07:00
parent 6b8a5b891d
commit 3057dbaf42
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/core.py
+7 -1
View File
@@ -20,6 +20,8 @@ opCLIPTextEncode = CLIPTextEncode()
opEmptyLatentImage = EmptyLatentImage() opEmptyLatentImage = EmptyLatentImage()
opVAEDecode = VAEDecode() opVAEDecode = VAEDecode()
cv2_is_top = False
class StableDiffusionModel: class StableDiffusionModel:
def __init__(self, unet, vae, clip, clip_vision): def __init__(self, unet, vae, clip, clip_vision):
@@ -62,6 +64,7 @@ def get_previewer(device, latent_format):
taesd = TAESD(None, taesd_decoder_path).to(device) taesd = TAESD(None, taesd_decoder_path).to(device)
def preview_function(x0, step, total_steps): def preview_function(x0, step, total_steps):
global cv2_is_top
with torch.no_grad(): with torch.no_grad():
x_sample = taesd.decoder(x0).detach() * 255.0 x_sample = taesd.decoder(x0).detach() * 255.0
x_sample = einops.rearrange(x_sample, 'b c h w -> b h w c') x_sample = einops.rearrange(x_sample, 'b c h w -> b h w c')
@@ -70,7 +73,9 @@ def get_previewer(device, latent_format):
flag = f'OpenCV Diffusion Preview {i}' flag = f'OpenCV Diffusion Preview {i}'
cv2.imshow(flag, s) cv2.imshow(flag, s)
cv2.setWindowTitle(flag, f'Preview Image {i} [{step}/{total_steps}]') cv2.setWindowTitle(flag, f'Preview Image {i} [{step}/{total_steps}]')
cv2.setWindowProperty(flag, cv2.WND_PROP_TOPMOST, 1) if not cv2_is_top:
cv2.setWindowProperty(flag, cv2.WND_PROP_TOPMOST, 1)
cv2_is_top = True
cv2.waitKey(1) cv2.waitKey(1)
taesd.preview = preview_function taesd.preview = preview_function
@@ -79,6 +84,7 @@ def get_previewer(device, latent_format):
def close_all_preview(): def close_all_preview():
cv2_is_top = False
cv2.destroyAllWindows() cv2.destroyAllWindows()