From f87956e9cf362b9a62bc9bb4911bcd23494f99a1 Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" Date: Sat, 19 Apr 2025 19:33:27 +0000 Subject: [PATCH] Apply style fixes --- .../schedulers/scheduling_flow_match_euler_discrete.py | 4 ++-- 1 file changed, 2 insertions(+), 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 0c1d6f5f8c..deedfbca40 100644 --- a/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py +++ b/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py @@ -443,10 +443,10 @@ class FlowMatchEulerDiscreteScheduler(SchedulerMixin, ConfigMixin): lower_mask = sigmas < per_token_sigmas[None] - 1e-6 lower_sigmas = lower_mask * sigmas lower_sigmas, _ = lower_sigmas.max(dim=0) - + current_sigma = per_token_sigmas[..., None] next_sigma = lower_sigmas[..., None] - dt = next_sigma - current_sigma # Equivalent to sigma_next - sigma + dt = next_sigma - current_sigma # Equivalent to sigma_next - sigma else: sigma_idx = self.step_index sigma = self.sigmas[sigma_idx]