From dcbb18a30ab06fb263897b045067f8884cf00a66 Mon Sep 17 00:00:00 2001 From: Robbin Marcus Date: Tue, 5 May 2026 04:37:16 +0200 Subject: [PATCH] Fix ignored generator in FlowMatchEulerDiscreteScheduler (#13678) --- .../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 1021abf0f6..7b207f7820 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 @@ import torch 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 @@ class FlowMatchEulerDiscreteScheduler(SchedulerMixin, ConfigMixin): 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