Skip to content

Wan checkpointing - #246

Merged
coolkp merged 6 commits into
mainfrom
wan-checkpointing-test
Sep 18, 2025
Merged

coolkp merged 6 commits into
mainfrom
wan-checkpointing-test

Conversation

@coolkp

@coolkp coolkp commented Sep 16, 2025

Copy link
Copy Markdown
Collaborator

Fixed issue of >flash_block_sizes in config or cli, causes saving the checkpoint to crash.
Changed behavior

  • Parse objects of dataclasses correctly
  • Skip with warning incompatible fields (this is harmless since checkpointing is for checkpointing mainly model config and weights and restore args for checkpoint only use that.)
    Added test for verifying.

@github-actions

Copy link
Copy Markdown

entrpn
entrpn previously approved these changes Sep 16, 2025
entrpn
entrpn previously approved these changes Sep 16, 2025
@coolkp
coolkp merged commit f41036d into main Sep 18, 2025
2 of 3 checks passed
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