Skip to content

[docs] Move diffusion model examples#3901

Merged
cuichenx merged 1 commit into
mainfrom
chcui/move-diffusion-examples-to-models
May 22, 2026
Merged

[docs] Move diffusion model examples#3901
cuichenx merged 1 commit into
mainfrom
chcui/move-diffusion-examples-to-models

Conversation

@cuichenx

Copy link
Copy Markdown
Contributor

Summary

  • Move Flux and Wan model-family examples from examples/diffusion/models to examples/models.
  • Update documentation and path references for the new example locations.

Validation

  • Stale path search for examples/diffusion/models, diffusion/models/(flux|wan), examples.diffusion.models, and examples/models/wan/conf returned no hits.
  • python3 -m py_compile passed for moved Flux/Wan Python examples plus touched Wan performance config and conversion functional tests.
  • git diff --check passed.
  • uv tool run ruff check ... passed.
  • uv tool run ruff format --check ... passed.
  • uv run pre-commit run --all-files was attempted but blocked on this host because locked nvidia-resiliency-ext==0.6.0 only has manylinux_2_39 wheels while the host resolves as manylinux_2_31_x86_64.

Blast Radius / Test Tier

  • Changed surface is examples/docs plus functional-test path constants only.
  • Recommend normal L0 Flux/Wan coverage.
  • L1/L2 not required for this move.

Signed-off-by: Chen Cui <chcui@nvidia.com>
@claude

claude Bot commented May 20, 2026

Copy link
Copy Markdown
Contributor

LGTM — clean file move with consistent path updates across docs, examples, and tests. No stale references to the old examples/diffusion/models/ path remain.

One pre-existing note (not introduced by this PR): examples/models/wan/README-perf-test.md lines 69 and 105 reference examples/models/wan/pretrain_wan.py, which does not exist (it did not exist at the old path either). Worth a follow-up cleanup.

Suggested test cases:

No perf tests impacted. The only perf config change (wan_workload_base_configs.py) is a docstring-only edit with no functional effect.

The following existing functional tests should be run to confirm the moved conversion scripts are found at their new paths:

  • test_flux_conversion (tests/functional_tests/test_groups/diffusion/flux/test_flux_conversion.py)
  • test_wan_conversion (tests/functional_tests/test_groups/diffusion/wan/test_wan_conversion.py)

@yaoyu-33 yaoyu-33 added area:diffusion DFM module docs Documentation-only updates or documentation debt needs-review PR is ready for code review and waiting on a reviewer labels May 20, 2026
@cuichenx cuichenx requested review from abhinavg4, huvunvidia and yaoyu-33 and removed request for abhinavg4 May 20, 2026 21:07

@huvunvidia huvunvidia left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Looks reasonable. Thank you!

@cuichenx cuichenx merged commit e4d2701 into main May 22, 2026
104 checks passed
@cuichenx cuichenx deleted the chcui/move-diffusion-examples-to-models branch May 22, 2026 18:10
vasunvidia pushed a commit to vasunvidia/Megatron-Bridge that referenced this pull request Jun 10, 2026
Signed-off-by: Chen Cui <chcui@nvidia.com>
Signed-off-by: Vasudevan Rengasamy <vrengasamy@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:diffusion DFM module docs Documentation-only updates or documentation debt needs-review PR is ready for code review and waiting on a reviewer

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants