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

wip base64

Browse Source
This commit is contained in:
damian 2022-11-01 11:40:40 +01:00 committed by Lincoln Stein
parent 9b984e0d1e
commit 62fe308f84
1 changed files with 8 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
backend/invoke_ai_web_server.py
View File

@ -615,9 +615,8 @@ class InvokeAIWebServer:
progress.set_current_status_has_steps(True) progress.set_current_status_has_steps(True)
if ( if (
generation_parameters["progress_images"] generation_parameters['progress_images'] and step % 5 == 0 \
and step % generation_parameters['save_intermediates'] == 0 and step < generation_parameters['steps'] - 1
and step < generation_parameters["steps"] - 1
): ):
image = self.generate.sample_to_image(sample) image = self.generate.sample_to_image(sample)
metadata = self.parameters_to_generated_image_metadata( metadata = self.parameters_to_generated_image_metadata(
@ -647,30 +646,25 @@ class InvokeAIWebServer:
"height": height, "height": height,
}, },
) )
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")
<<<<<<< HEAD img_str = base64.b64encode(buffered.getvalue())
img_base64 = "data:image/jpeg;base64," + base64.b64encode(buffered.getvalue()).decode('UTF-8') img_base64 = bytes("data:image/jpeg;base64,", encoding='utf-8') + img_str
=======
img_str = base64.b64encode(buffered.getvalue()).decode('UTF-8')
img_base64 = "data:image/jpeg;base64," + img_str
>>>>>>> 4013e8a... Fixes b64 image sending and displaying
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())
eventlet.sleep(0) eventlet.sleep(0)