:memo: Change

app.py

@@ -79,6 +79,7 @@ def inference(
         num_images: int = 4,
         safety_checker: bool = True,
         use_model_offload:bool = False,
+        seed: int = 8888,
         progress=gr.Progress(track_tqdm=True),
 ) -> Image.Image:
     progress(0, "Starting inference...")
@@ -113,6 +114,8 @@ def inference(
     else:
         pipe = pipe.to('cuda')
 
+    generator = torch.Generator(device=device).manual_seed(seed)
+
     images = pipe(
         prompt,
         negative_prompt=negative_prompt,
@@ -121,6 +124,7 @@ def inference(
         guidance_scale=guidance_scale,
         num_inference_steps=num_inference_steps,
         num_images_per_prompt=num_images,
+        generator=generator,
     ).images
 
     if num_images % 2 == 1:
@@ -155,6 +159,7 @@ if __name__ == "__main__":
                         width = gr.Number(label="Width", value=512, step=64, minimum=64, maximum=2048)
                         height = gr.Number(label="Height", value=512, step=64, minimum=64, maximum=2048)
                         num_images = gr.Number(label="Num Images", value=4, minimum=1, maximum=10, step=1)
+                        seed = gr.Number(label="Seed", value=8888,  step=1)
 
                     guidance_scale = gr.Slider(label="Guidance Scale", value=7.5, step=0.5, minimum=0, maximum=10)
                     num_inference_step = gr.Slider(
@@ -179,6 +184,7 @@ if __name__ == "__main__":
             num_images,
             use_safety_checker,
             use_model_offload,
+            seed,
         ]
 
         btn = gr.Button("Generate")