Skip to content

fix(ltx2): restore Stage 1 sigma schedule at each run start#963

Merged
helloyongyang merged 1 commit intomainfrom
fix/ltx2-restore-stage1-sigmas-on-init-run
Mar 30, 2026
Merged

fix(ltx2): restore Stage 1 sigma schedule at each run start#963
helloyongyang merged 1 commit intomainfrom
fix/ltx2-restore-stage1-sigmas-on-init-run

Conversation

@xinyiqin
Copy link
Copy Markdown
Contributor

Restores the full Stage 1 sigma schedule in init_run() before each job so long-lived workers do not keep the upsampler's shorter sigma schedule (which left Stage 1 at 3 steps instead of 8 after a prior Stage 2 run).

Copy link
Copy Markdown
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request updates the init_run method in ltx2_runner.py to support custom sigma values. It introduces a check for distilled_sigma_values in the configuration and resets the model scheduler's sigmas accordingly if they are provided. I have no feedback to provide.

@XHPlus XHPlus force-pushed the fix/ltx2-restore-stage1-sigmas-on-init-run branch from 2c5fd4c to 88eece3 Compare March 27, 2026 18:04
@helloyongyang helloyongyang merged commit 38f861a into main Mar 30, 2026
2 checks passed
@helloyongyang helloyongyang deleted the fix/ltx2-restore-stage1-sigmas-on-init-run branch March 30, 2026 03:32
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