Skip to content

fix: ensure custom timesteps are consistent with sigmas in FlowMatchEulerDiscreteScheduler#14036

Open
KartavyaDikshit wants to merge 2 commits into
huggingface:mainfrom
KartavyaDikshit:fix-issue-14013
Open

fix: ensure custom timesteps are consistent with sigmas in FlowMatchEulerDiscreteScheduler#14036
KartavyaDikshit wants to merge 2 commits into
huggingface:mainfrom
KartavyaDikshit:fix-issue-14013

Conversation

@KartavyaDikshit

Copy link
Copy Markdown

Resolves #14013

When custom timesteps were provided to set_timesteps with shift != 1, the timesteps array would diverge from the sigmas array after internal computation. This fix ensures custom timesteps are processed consistently with the shift factor, matching the behavior of auto-generated timesteps.

Signed-off-by: KartavyaDikshit kartavyaniraj.dikshit2021@vitstudent.ac.in

KartavyaDikshit added 2 commits June 22, 2026 11:05
…ulerDiscreteScheduler

Signed-off-by: KartavyaDikshit <kartavyaniraj.dikshit2021@vitstudent.ac.in>
…ulerDiscreteScheduler

Signed-off-by: KartavyaDikshit <kartavyaniraj.dikshit2021@vitstudent.ac.in>
@github-actions github-actions Bot added fixes-issue size/L PR with diff > 200 LOC schedulers and removed size/L PR with diff > 200 LOC labels Jun 22, 2026
@TheLovesOfLadyPurple

Copy link
Copy Markdown

I already committed a pr there: #14011
The method I use is totally identical to yours.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

A bug on the set_timesteps function of the FlowMatchEulerDiscreteScheduler

2 participants