From b109af99502dc923eb7bd3f5e9fc6fe0cfd53475 Mon Sep 17 00:00:00 2001
From: regisss <15324346+regisss@users.noreply.github.com>
Date: Wed, 17 Jan 2024 17:44:46 +0000
Subject: [PATCH] Move `set_seed` in SD example

---
 examples/stable-diffusion/text_to_image_generation.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/examples/stable-diffusion/text_to_image_generation.py b/examples/stable-diffusion/text_to_image_generation.py
index 088279d706..0526c1ce60 100755
--- a/examples/stable-diffusion/text_to_image_generation.py
+++ b/examples/stable-diffusion/text_to_image_generation.py
@@ -236,15 +236,16 @@ def main():
     if args.bf16:
         kwargs["torch_dtype"] = torch.bfloat16
 
-    # Set seed before running the model
-    set_seed(args.seed)
-
     # Generate images
     if sdxl:
         pipeline = GaudiStableDiffusionXLPipeline.from_pretrained(
             args.model_name_or_path,
             **kwargs,
         )
+
+        # Set seed before running the model
+        set_seed(args.seed)
+
         outputs = pipeline(
             prompt=args.prompts,
             prompt_2=args.prompts_2,
@@ -263,6 +264,10 @@ def main():
             args.model_name_or_path,
             **kwargs,
         )
+
+        # Set seed before running the model
+        set_seed(args.seed)
+
         outputs = pipeline(
             prompt=args.prompts,
             num_images_per_prompt=args.num_images_per_prompt,