Fix sequence padding for DiT. Support CP + THD for DiT#39

Merged

abhinavg4 merged 20 commits intomainfrom

Nov 17, 2025

Commits on Nov 13, 2025

Commits on Nov 14, 2025

Implement ProcessGroupCollection initialization in DiT and Wan models
abhinavg4
committed
Merge branch 'main' into megatron_fixes
abhinavg4
committed
Update CONTRIBUTING.md to include detailed setup instructions for development environment and Docker container usage. Added sections for building and running the container, as well as setting the P…
abhinavg4
committed
Refactor import statements in dit_model.py to streamline dependencies. Removed redundant import of ProcessGroupCollection, enhancing code clarity and maintainability.
abhinavg4
committed
Refactor code style in DiT and Wan models
abhinavg4
committed

Commits on Nov 15, 2025

Commits on Nov 17, 2025