huggingface/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-06-02 00:01:34 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Fix ignored generator in FlowMatchEulerDiscreteScheduler (#13678)

Browse Source
This commit is contained in:
Robbin Marcus
2026-05-05 04:37:16 +02:00
committed by GitHub
parent c8eba433ad
commit dcbb18a30a
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py
View File

@@ -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