Skip to content

feat: Add support for flux#94

Merged
pthombre merged 23 commits intomainfrom
pranav/flux_support_v1
Jan 30, 2026
Merged

feat: Add support for flux#94
pthombre merged 23 commits intomainfrom
pranav/flux_support_v1

Conversation

@pthombre
Copy link
Contributor

@pthombre pthombre commented Jan 27, 2026 •
edited
Loading

PR adds:

  1. Flux adapter class required for running the Flow Matching Pipeline
  2. Generalizes the AutoDiffusionClass to work of yaml config, instead of defining individual classes for each model
  3. Data processor to generate training data for the multi-resolution dataloader. Currently only Flux data generation is supported. This will be generalized for more model types.

pthombre and others added 6 commits January 21, 2026 15:43
@pthombre
feat: Add DDP support in the Automodel path
ff8d0c3 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Fix linting issues
5d246a4 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Merge branch 'main' into pranav/ddp_support
9edbc34
@linnanwang
add cicd nightly tests
4ab7115 
Signed-off-by: linnan wang <linnanw@nvidia.com>
@linnanwang
add cicd nightly tests
ef14d82 
Signed-off-by: linnan wang <linnanw@nvidia.com>
@pthombre
feat: Flux training support
3b2d3a8 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@copy-pr-bot
Copy link

copy-pr-bot bot commented Jan 27, 2026

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
Merge branch 'main' into pranav/flux_support_v1
cdd5f56
@pthombre
Formatting changes
52e684b 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Changing import order
dfb8608 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Linting fix
507e453 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Fixing imports
d136c87 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Init fix
85b4923 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Code fixes
0c2dea8 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Import order fix
cf44aa0 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Fix linting
cf40333 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Copy link
Contributor Author

pthombre commented Jan 28, 2026

/ok to test cf40333

@pthombre pthombre changed the title Pranav/flux support v1 feat: Add support for flux Jan 28, 2026
@pthombre pthombre marked this pull request as ready for review January 28, 2026 01:21
@pthombre pthombre requested a review from a team as a code owner January 28, 2026 01:21
@pthombre
Fix broken unit tests
c479f27 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Fix lint issue
025f45b 
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre
Merge branch 'main' into pranav/flux_support_v1
d4b01f3
@pthombre
Copy link
Contributor Author

pthombre commented Jan 28, 2026

/ok to test d4b01f3

@pthombre pthombre requested a review from akoumpa January 29, 2026 01:10
akoumpa
akoumpa previously approved these changes Jan 30, 2026
View reviewed changes
ko3n1g
ko3n1g previously approved these changes Jan 30, 2026
View reviewed changes
pthombre and others added 2 commits January 30, 2026 09:59
@pthombre
Merge branch 'main' into pranav/flux_support_v1
4ea3ab4
@chtruong814
Update secrets baseline
0641f5a 
Signed-off-by: Charlie Truong <chtruong@nvidia.com>
@chtruong814 chtruong814 dismissed stale reviews from ko3n1g and akoumpa via 0641f5a January 30, 2026 18:51
@chtruong814
Copy link
Collaborator

chtruong814 commented Jan 30, 2026

/ok to test 0641f5a

@pthombre pthombre enabled auto-merge (squash) January 30, 2026 19:45
@pthombre
Copy link
Contributor Author

pthombre commented Jan 30, 2026

/ok to test 02a4f39

@pthombre pthombre merged commit 5b81246 into main Jan 30, 2026
19 checks passed
@pthombre pthombre deleted the pranav/flux_support_v1 branch January 30, 2026 21:50
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@chtruong814 chtruong814 chtruong814 approved these changes

@ko3n1g ko3n1g ko3n1g left review comments

@akoumpa akoumpa akoumpa left review comments

@linnanwang linnanwang Awaiting requested review from linnanwang

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 @chtruong814 @ko3n1g @akoumpa @linnanwang