Skip to content

Conversation

@bghira
Copy link
Owner

@bghira bghira commented Feb 1, 2026

This pull request makes a targeted change to the logic for determining whether to apply a dynamic shift when getting the timestep schedule in validation.py. The update tightens the condition so that both flow_schedule_auto_shift and use_dynamic_shifting must be enabled, rather than either one.

  • Logic correction for dynamic shift application:
    • In get_timestep_schedule, the condition now requires both self.config.flow_schedule_auto_shift and dynamic_shift (from noise_scheduler.config.use_dynamic_shifting) to be true before applying the dynamic shift logic, instead of just one or the other. This prevents unintended application of dynamic shifting when only one flag is set.

@bghira bghira merged commit 12b156f into main Feb 1, 2026
2 checks passed
@bghira bghira deleted the bugfix/validation-for-shiftless-models branch February 1, 2026 15:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants