1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00

Apply style fixes

This commit is contained in:
github-actions[bot]
2025-04-19 19:33:27 +00:00
parent 690adb5bd9
commit f87956e9cf

View File

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