github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-18 16:14:50 -05:00
Code Issues Packages Projects Releases Wiki Activity

add option to show intermediate latent space

Browse Source
This commit is contained in:
damian
2022-11-01 11:17:43 +01:00
committed by Lincoln Stein
parent be1393a41c
commit cdb107dcda
9 changed files with 751 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
backend/invoke_ai_web_server.py
View File

@@ -604,12 +604,15 @@ class InvokeAIWebServer:
progress.set_current_status("Generating")
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 (
generation_parameters["progress_images"]
and step % 5 == 0
and step < generation_parameters["steps"] - 1
wants_progress_image | wants_progress_latents
and step < generation_parameters['steps'] - 1
):
image = self.generate.sample_to_image(sample)
image = self.generate.sample_to_image(sample) if wants_progress_image \
else self.generate.sample_to_lowres_estimated_image(sample)
metadata = self.parameters_to_generated_image_metadata(
generation_parameters
)