Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

webgui working again

Browse Source
This commit is contained in:
Lincoln Stein 2022-11-02 18:07:18 -04:00
parent 76249b3d4e
commit 09625eae66
1 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
backend/invoke_ai_web_server.py
View File

@ -614,15 +614,12 @@ class InvokeAIWebServer:
progress.set_current_status("Generating") progress.set_current_status("Generating")
progress.set_current_status_has_steps(True) progress.set_current_status_has_steps(True)
wants_progress_image = generation_parameters['progress_images'] and step % 5 == 0
wants_progress_latents = generation_parameters['progress_latents']
if ( if (
wants_progress_image | wants_progress_latents generation_parameters["progress_images"]
and step < generation_parameters['steps'] - 1 and step % generation_parameters['save_intermediates'] == 0
and step < generation_parameters["steps"] - 1
): ):
image = self.generate.sample_to_image(sample) if wants_progress_image \ image = self.generate.sample_to_image(sample)
else self.generate.sample_to_lowres_estimated_image(sample)
metadata = self.parameters_to_generated_image_metadata( metadata = self.parameters_to_generated_image_metadata(
generation_parameters generation_parameters
) )
@ -651,22 +648,26 @@ class InvokeAIWebServer:
}, },
) )
if generation_parameters['progress_latents']: if generation_parameters["progress_latents"]:
image = self.generate.sample_to_lowres_estimated_image(sample) image = self.generate.sample_to_lowres_estimated_image(sample)
(width, height) = image.size (width, height) = image.size
width *= 8
height *= 8
buffered = io.BytesIO() buffered = io.BytesIO()
image.save(buffered, format="PNG") image.save(buffered, format="PNG")
img_str = base64.b64encode(buffered.getvalue()) img_base64 = "data:image/png;base64," + base64.b64encode(
img_base64 = bytes("data:image/jpeg;base64,", encoding='utf-8') + img_str buffered.getvalue()
).decode("UTF-8")
self.socketio.emit( self.socketio.emit(
"intermediateResult", "intermediateResult",
{ {
"url": img_base64, "url": img_base64,
"isBase64": True,
"mtime": 0, "mtime": 0,
"metadata": {}, "metadata": {},
"width": width, "width": width,
"height": height, "height": height,
} },
) )
self.socketio.emit("progressUpdate", progress.to_formatted_dict()) self.socketio.emit("progressUpdate", progress.to_formatted_dict())