[SDXL] finish

patrickvonplaten · Oct 3, 2023 · 163494f · 163494f
1 parent 500cb89
commit 163494f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/examples/research_projects/sdxl_flax/sdxl_single.py b/examples/research_projects/sdxl_flax/sdxl_single.py
@@ -101,3 +101,6 @@ def generate(
 neg_prompt = "cartoon, illustration, animation. face. male, female"
 images = generate(prompt, neg_prompt)
 print(f"Inference in {time.time() - start}")
+
+for i, image in enumerate(images):
+    image.save(f"castle_{i}.png")
diff --git a/examples/research_projects/sdxl_flax/sdxl_single_aot.py b/examples/research_projects/sdxl_flax/sdxl_single_aot.py
@@ -138,3 +138,6 @@ def generate(prompt, negative_prompt, seed=default_seed, guidance_scale=default_
 neg_prompt = "cartoon, illustration, animation. face. male, female"
 images = generate(prompt, neg_prompt)
 print(f"Inference in {time.time() - start}")
+
+for i, image in enumerate(images):
+    image.save(f"castle_{i}.png")