Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-08 22:08:42 +02:00
Code Issues Packages Projects Releases Wiki Activity

add latent history

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-10-08 09:23:16 -04:00
parent a3a177277a
commit 2e08a26d53
13 changed files with 145 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_diffusers.py
+4 -2
View File
@@ -95,6 +95,7 @@ def process_base(p: processing.StableDiffusionProcessing):
output = shared.sd_model(**base_args)
if isinstance(output, dict):
output = SimpleNamespace(**output)
shared.history.add(output.images, info=processing.create_infotext(p), ops=p.ops)
timer.process.record('pipeline')
hidiffusion.unapply()
sd_models_compile.openvino_post_compile(op="base") # only executes on compiled vino models
@@ -209,6 +210,7 @@ def process_hires(p: processing.StableDiffusionProcessing, output):
output = shared.sd_model(**hires_args) # pylint: disable=not-callable
if isinstance(output, dict):
output = SimpleNamespace(**output)
shared.history.add(output.images, info=processing.create_infotext(p), ops=p.ops)
sd_models_compile.check_deepcache(enable=False)
sd_models_compile.openvino_post_compile(op="base")
except AssertionError as e:
@@ -281,6 +283,7 @@ def process_refine(p: processing.StableDiffusionProcessing, output):
output = shared.sd_refiner(**refiner_args) # pylint: disable=not-callable
if isinstance(output, dict):
output = SimpleNamespace(**output)
shared.history.add(output.images, info=processing.create_infotext(p), ops=p.ops)
sd_models_compile.openvino_post_compile(op="refiner")
except AssertionError as e:
shared.log.info(e)
@@ -322,7 +325,6 @@ def process_decode(p: processing.StableDiffusionProcessing, output):
full_quality = p.full_quality,
width = width,
height = height,
save = p.state == '',
)
elif hasattr(output, 'images'):
results = output.images
@@ -388,7 +390,7 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
if 'base' not in p.skip:
output = process_base(p)
else:
output = SimpleNamespace(images=shared.history.latest)
output, _index = SimpleNamespace(images=shared.history.selected)
if shared.state.interrupted or shared.state.skipped:
shared.sd_model = orig_pipeline