github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-09 13:57:54 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix slow conversion to image in SD web gui (#586)

Browse Source
This commit is contained in:
Quinn Dawkins
2022-12-08 00:35:51 -05:00
committed by GitHub
parent 9a14298146
commit dab9403b8f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
web/models/stable_diffusion/main.py
View File

@@ -106,8 +106,8 @@ def stable_diff_inf(
latents_numpy = latents.detach().numpy()
image = vae.forward((latents_numpy,))
image = torch.from_numpy(image)
image = image.detach().cpu().permute(0, 2, 3, 1).numpy()
images = (image * 255).round().astype("uint8")
image = (image.detach().cpu().permute(0, 2, 3, 1) * 255.0).numpy()
images = image.round().astype("uint8")
pil_images = [Image.fromarray(image) for image in images]
out_img = pil_images[0]