Skip to content

feat:Adding Automodel support for Wan 2.1 #18

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
akoumpa merged 20 commits into from
Oct 31, 2025
Merged

feat:Adding Automodel support for Wan 2.1 #18

akoumpa merged 20 commits into from
Oct 31, 2025

Conversation

@pthombre
Copy link
Contributor

@pthombre pthombre commented Oct 28, 2025

No description provided.

@pthombre
feat:Adding Automodel support for Wan 2.1
f32e717 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Add fixes to ensure that required dependencies are fetched from dfm
a7fbb34 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Adding checkpointer object in training recipe
c140bcd 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@copy-pr-bot
Copy link

copy-pr-bot bot commented Oct 28, 2025

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@pthombre
Migrate Autodiffusion pipeline to DFM
19fdada 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Move FSDP manager creation into NeMo Autodiffusion Pipeline
a104de3 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre pthombre marked this pull request as ready for review October 28, 2025 23:44
@pthombre
Add missing copyright to file
c86f6bd 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Fixing lint errors
b4324d0 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Move wandb import
0498474 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Removing debug comment
394eb99 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
linnanwang
linnanwang previously approved these changes Oct 29, 2025
View reviewed changes
Copy link
Contributor

@linnanwang linnanwang left a comment

Choose a reason for hiding this comment

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

LGTM

@linnanwang
Copy link
Contributor

linnanwang commented Oct 29, 2025

@abhinavg4 @akoumpa please take a look as well. Thanks.

akoumpa
akoumpa reviewed Oct 29, 2025
View reviewed changes
@pthombre
Adding dataset files for Wan
7a2715a 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
pthombre and others added 4 commits October 29, 2025 15:25
@pthombre
Fix linting
4303eae 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pablo-garay
init: add dependencies + uv.lock
c4d3aba
@pthombre
Fix bug in Wan dataloader
0be838d 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@akoumpa
Merge pull request #20 from NVIDIA-NeMo/teamwork-groupforce/dependenc…
6104c9f 
…y-add

feat: update pyproject.toml
@akoumpa akoumpa requested a review from a team as a code owner October 30, 2025 05:08
@pthombre
Add fix for multinode deployment
dc05842 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Add multinode recipe
9ad8c86 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
linnanwang
linnanwang previously approved these changes Oct 30, 2025
View reviewed changes
Copy link
Contributor

@linnanwang linnanwang left a comment

Choose a reason for hiding this comment

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

all things LGTM

@pthombre
Modifying Automodel dependencies
77bd4c6 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
linting fixes
39ac355 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre pthombre requested a review from akoumpa October 30, 2025 20:17
akoumpa
akoumpa approved these changes Oct 31, 2025
View reviewed changes
Copy link
Contributor

@akoumpa akoumpa left a comment

Choose a reason for hiding this comment

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

LGTM, thank you.

@akoumpa akoumpa merged commit 23d4a67 into main Oct 31, 2025
4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@pablo-garay pablo-garay pablo-garay approved these changes

@akoumpa akoumpa akoumpa approved these changes

@linnanwang linnanwang linnanwang left review comments

@abhinavg4 abhinavg4 Awaiting requested review from abhinavg4

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

5 participants

@pthombre @linnanwang @pablo-garay @akoumpa