Skip to content

Commit 25bc77d

Browse files
apolinarioyiyixuxu
andauthored
Update src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py
Co-authored-by: YiYi Xu <yixu310@gmail.com>
1 parent 9c35a89 commit 25bc77d

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -404,8 +404,6 @@ def step(
404404
A random number generator.
405405
per_token_timesteps (`torch.Tensor`, *optional*):
406406
The timesteps for each token in the sample.
407-
stochastic_sampling (`bool`, *optional*):
408-
Whether to use stochastic sampling. If None, defaults to the value set in the scheduler's config.
409407
return_dict (`bool`):
410408
Whether or not to return a
411409
[`~schedulers.scheduling_flow_match_euler_discrete.FlowMatchEulerDiscreteSchedulerOutput`] or tuple.

0 commit comments

Comments
 (0)