From 6c4ab4a09b07b6c197b8e54aa1bdd55b67b06f72 Mon Sep 17 00:00:00 2001
From: Alexander Pivovarov <apivovarov@gmail.com>
Date: Tue, 20 Jun 2023 15:24:50 -0700
Subject: [PATCH] Save all generated images in SD demo_alt.py

---
 examples/05_stable_diffusion/scripts/demo_alt.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/examples/05_stable_diffusion/scripts/demo_alt.py b/examples/05_stable_diffusion/scripts/demo_alt.py
index 28b322f02..ebd4c3798 100644
--- a/examples/05_stable_diffusion/scripts/demo_alt.py
+++ b/examples/05_stable_diffusion/scripts/demo_alt.py
@@ -49,15 +49,16 @@ def run(
     prompt = [prompt] * batch
     negative_prompt = [negative_prompt] * batch
     with torch.autocast("cuda"):
-        image = pipe(
+        images = pipe(
             prompt=prompt,
             height=height,
             width=width,
             negative_prompt=negative_prompt,
             num_inference_steps=steps,
             guidance_scale=cfg,
-        ).images[0]
-    image.save("example_ait.png")
+        ).images
+    for i, image in enumerate(images):
+        image.save(f"example_ait_{i}.png")
 
 
 if __name__ == "__main__":