From 25bc77d8f88ec970916e66c81854e2b2371ceb2b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?apolin=C3=A1rio?= Date: Mon, 21 Apr 2025 23:34:24 +0200 Subject: [PATCH] Update src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py Co-authored-by: YiYi Xu --- .../schedulers/scheduling_flow_match_euler_discrete.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py b/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py index f622f4217c..b1eddac5b0 100644 --- a/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py +++ b/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py @@ -404,8 +404,6 @@ class FlowMatchEulerDiscreteScheduler(SchedulerMixin, ConfigMixin): A random number generator. per_token_timesteps (`torch.Tensor`, *optional*): The timesteps for each token in the sample. - stochastic_sampling (`bool`, *optional*): - Whether to use stochastic sampling. If None, defaults to the value set in the scheduler's config. return_dict (`bool`): Whether or not to return a [`~schedulers.scheduling_flow_match_euler_discrete.FlowMatchEulerDiscreteSchedulerOutput`] or tuple.