Update stable_cascade.py

stable_cascade.py

@@ -13,6 +13,7 @@ def generate_images(
     height=1024,
     width=1024, 
     guidance_scale=4.0, 
+    seed=42,
     num_images_per_prompt=1,
     prior_inference_steps=20, 
     decoder_inference_steps=10
@@ -30,10 +31,12 @@ def generate_images(
     Returns:
     - List[PIL.Image]: A list of generated PIL Image objects.
     """
+    generator = torch.Generator(device="cuda").manual_seed(seed)
 
     # Generate image embeddings using the prior model
     prior_output = prior(
         prompt=prompt,
+        generator=generator,
         height=height,
         width=width,
         negative_prompt=negative_prompt,
@@ -46,6 +49,7 @@ def generate_images(
     decoder_output = decoder(
         image_embeddings=prior_output.image_embeddings.half(),
         prompt=prompt,
+        generator=generator,
         negative_prompt=negative_prompt,
         guidance_scale=0.0,  # Guidance scale typically set to 0 for decoder as guidance is applied in the prior
         output_type="pil",
@@ -70,6 +74,15 @@ def web_demo():
                     placeholder="Negative Prompt",
                     show_label=False,
                 )
+                
+                text2image_num_images_per_prompt = gr.Slider(
+                    minimum=1,
+                    maximum=1000000,
+                    step=10,
+                    value=42,
+                    label="Seed",
+                )
+                
                 with gr.Row():
                     with gr.Column():
                         text2image_num_images_per_prompt = gr.Slider(