From b1abfb4cc5e22cdd24317e98eb1b003c580b3cbe Mon Sep 17 00:00:00 2001
From: Robbin Marcus <4852449+RobbinMarcus@users.noreply.github.com>
Date: Sun, 3 May 2026 19:56:52 +0200
Subject: [PATCH] Fix ignored generator in FlowMatchEulerDiscreteScheduler

---
 .../schedulers/scheduling_flow_match_euler_discrete.py         | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py b/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py
index 1021abf0f6f6..7b207f782079 100644
--- a/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py
+++ b/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py
@@ -21,6 +21,7 @@
 
 from ..configuration_utils import ConfigMixin, register_to_config
 from ..utils import BaseOutput, is_scipy_available, logging
+from ..utils.torch_utils import randn_tensor
 from .scheduling_utils import SchedulerMixin
 
 
@@ -507,7 +508,7 @@ def step(
 
         if self.config.stochastic_sampling:
             x0 = sample - current_sigma * model_output
-            noise = torch.randn_like(sample)
+            noise = randn_tensor(sample.shape, generator=generator, device=sample.device, dtype=sample.dtype)
             prev_sample = (1.0 - next_sigma) * x0 + next_sigma * noise
         else:
             prev_sample = sample + dt * model_output