Update app.py

app.py

@@ -956,8 +956,9 @@ class SDXLFlowMatchingPipeline:
                 dt = -1.0 / num_inference_steps
                 latents = latents + dt * noise_pred
             else:
+                # Pass generator for deterministic ancestral/SDE sampling
                 latents = self.scheduler.step(
-                    noise_pred, t, latents, return_dict=False
+                    noise_pred, t, latents, generator=generator, return_dict=False
                 )[0]
         
         # Decode
@@ -1237,7 +1238,10 @@ class SD15FlowMatchingPipeline:
                 dt = -1.0 / num_inference_steps
                 latents = latents + dt * noise_pred
             else:
-                latents = self.scheduler.step(noise_pred, t, latents, return_dict=False)[0]
+                # Pass generator for deterministic ancestral/SDE sampling
+                latents = self.scheduler.step(
+                    noise_pred, t, latents, generator=generator, return_dict=False
+                )[0]
         
         latents = latents / self.vae_scale_factor