This commit is contained in:
Disty0
2025-10-30 00:52:43 +03:00
parent 9533b7e847
commit f7fef1f5b5
2 changed files with 9 additions and 2 deletions
+3 -1
View File
@@ -183,6 +183,8 @@ def process_base(p: processing.StableDiffusionProcessing):
output = SimpleNamespace(**output)
if isinstance(output, list):
output = SimpleNamespace(images=output)
if isinstance(output, Image.Image):
output = SimpleNamespace(images=[output])
if hasattr(output, 'images'):
shared.history.add(output.images, info=processing.create_infotext(p), ops=p.ops)
timer.process.record('pipeline')
@@ -196,7 +198,7 @@ def process_base(p: processing.StableDiffusionProcessing):
else:
shared.log.debug(f'Generated: frames={len(output.frames[0])}')
output.images = output.frames[0]
if isinstance(output.images, np.ndarray):
if hasattr(output, 'images') and isinstance(output.images, np.ndarray):
output.images = torch.from_numpy(output.images)
except AssertionError as e:
shared.log.info(e)