extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-04-22 18:38:38 +02:00
Code Issues Packages Projects Releases Wiki Activity

Typo in decoding live preview images

Browse Source
This commit is contained in:
cmdr2 2022-12-17 15:59:09 +05:30
parent aa8b50280b
commit 8189b38e6e
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ui/sd_internal/renderer.py
View File

@ -99,15 +99,14 @@ def make_step_callback(req: GenerateImageRequest, task_data: TaskData, data_queu
def update_temp_img(x_samples, task_temp_images: list): def update_temp_img(x_samples, task_temp_images: list):
partial_images = [] partial_images = []
for i in range(req.num_outputs): images = image_utils.latent_samples_to_images(context, x_samples)
img = image_utils.latent_to_img(context, x_samples[i].unsqueeze(0)) for i, img in enumerate(images):
buf = image_utils.img_to_buffer(img, output_format='JPEG') buf = image_utils.img_to_buffer(img, output_format='JPEG')
del img
context.temp_images[f"{task_data.request_id}/{i}"] = buf context.temp_images[f"{task_data.request_id}/{i}"] = buf
task_temp_images[i] = buf task_temp_images[i] = buf
partial_images.append({'path': f"/image/tmp/{task_data.request_id}/{i}"}) partial_images.append({'path': f"/image/tmp/{task_data.request_id}/{i}"})
del images
return partial_images return partial_images
def on_image_step(x_samples, i): def on_image_step(x_samples, i):