fix restore pipeline

This commit is contained in:
Vladimir Mandic
2024-02-17 07:53:03 -05:00
parent 8ae3b5bb43
commit 2a10875bd5
4 changed files with 16 additions and 14 deletions
+4
View File
@@ -316,6 +316,10 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
def infotext(index): # pylint: disable=function-redefined # noqa: F811
return create_infotext(p, p.prompts, p.seeds, p.subseeds, index=index, all_negative_prompts=p.negative_prompts)
if hasattr(shared.sd_model, 'restore_pipeline') and shared.sd_model.restore_pipeline is not None:
print('HERE')
shared.sd_model.restore_pipeline()
for i, x_sample in enumerate(x_samples_ddim):
p.batch_index = i
if type(x_sample) == Image.Image: