Pin nemo_automodel below 0.6 for the fastgen example by kevalmorabia97 · Pull Request #2260 · NVIDIA/Model-Optimizer

kevalmorabia97 · 2026-08-26T23:09:09Z

What does this PR do?

Type of change: Bug fix

nemo_automodel 0.6.0 removed nemo_automodel.recipes.diffusion.train.is_main_process without a replacement (it was a three-line rank-zero predicate in 0.5.0, and 0.6.0 defines no equivalent anywhere in the package). examples/diffusers/fastgen/dmd2_recipe.py imports it, so the example's import guard fires and every test in tests/examples/diffusers/ errors at collection:

ImportError: cannot import name 'is_main_process' from 'nemo_automodel.recipes.diffusion.train'
tests/examples/diffusers/fastgen/test_resume_dataloader.py
E ImportError: The DMD2 fastgen example requires `nemo_automodel`. ...
collected 42 items / 1 error

The requirement was >=0.4.0,<1.0, so CI picked 0.6.0 as soon as it was published and the onnx (diffusers) job started failing on every PR (e.g. runs 33020467654, 33019418460, 33010815298, 33007613265, 33006944292 — all unrelated branches). Capping at <0.6 restores the tested range.

Every other nemo_automodel symbol the example imports still exists in 0.6.0 (_diffusers.auto_diffusion_pipeline.NeMoAutoDiffusionPipeline, recipes.diffusion.train.TrainDiffusionRecipe, and the four components.datasets.diffusion.* helpers), so is_main_process is the only blocker; the alternative is defining that predicate locally and widening the cap again, which is worth doing separately if the example is meant to track 0.6.

Usage

pip install -r examples/diffusers/fastgen/requirements.txt

Testing

Reproduced the break by diffing the published wheels: is_main_process is defined at nemo_automodel/recipes/diffusion/train.py:692 in 0.5.0 and absent from 0.6.0 (grep -rn "def is_main_process" over the unpacked 0.6.0 wheel returns nothing). Confirmed the remaining imported symbols are all still present in 0.6.0.

CI on this PR exercises the fix directly: the onnx (diffusers) job installs from this requirements file and is the job that has been failing.

Before your PR is "Ready for review"

Is this change backward compatible?: ✅
If you copied code from any other sources or added a new PIP dependency, did you follow guidance in CONTRIBUTING.md: N/A — existing dependency, tightened bound.
Did you write any new necessary tests?: N/A — the existing tests/examples/diffusers/ suite is what this unblocks.
Did you update Changelog?: N/A — dependency-pin fix for a break introduced and fixed within the same unreleased cycle.
Did you get Claude approval on this PR?: ❌

🤖 Generated with Claude Code

Summary by CodeRabbit

Bug Fixes
- Fixed dependency compatibility for the FastGen diffusion example.
- Prevented installation of versions that could cause the example to fail at startup.

0.6.0 removed nemo_automodel.recipes.diffusion.train.is_main_process without a replacement, so dmd2_recipe.py fails to import and every diffusers example test errors at collection. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com> Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>

coderabbitai · 2026-08-26T23:09:59Z

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: c88c6187-1b4b-44f2-9939-04a9376fe97d

📥 Commits

Reviewing files that changed from the base of the PR and between a0513f1 and b2924a0.

📒 Files selected for processing (1)

examples/diffusers/fastgen/requirements.txt

Included review availability: Your plan provides up to 12 included reviews per hour; 7 remain after this review.

📝 Walkthrough

Walkthrough

The FastGen requirements file changes the nemo_automodel[diffusion] upper version bound from <1.0 to <0.6 while retaining the >=0.4.0 lower bound.

Changes

FastGen dependency compatibility

Layer / File(s)	Summary
Constrain diffusion dependency version `examples/diffusers/fastgen/requirements.txt`	The `nemo_automodel[diffusion]` requirement now allows versions from `0.4.0` up to, but not including, `0.6.0`.

Estimated code review effort: 1 (Trivial) | ~2 minutes

Merge Risk:⚪ Minimal · up to b2924

This localized dependency pin restores compatibility for the fastgen example, with no actionable merge-blocking risk remaining after normal checks and review.

Suggested reviewers:jingyu-ml, chenhanyu, shengliangxu

🚥 Pre-merge checks | ✅ 6

✅ Passed checks (6 passed)

Check name	Status	Explanation
Description Check	✅ Passed	Check skipped - CodeRabbit’s high-level summary is enabled.
Title check	✅ Passed	The title clearly and concisely describes the main change: pinning `nemo_automodel` below version 0.6 for the fastgen example.
Docstring Coverage	✅ Passed	No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check. Docstring coverage is scoped to functions touched by this diff. Analyzed 0 functions across 0…
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Security Anti-Patterns	✅ Passed	PASS. The commit changes only `examples/diffusers/fastgen/requirements.txt`; no `modelopt` or examples Python files changed. It replaces the existing `nemo_automodel[diffusion]` upper bound `<1.0` wit…

Full details: Docstring Coverage

Explanation

No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check. Docstring coverage is scoped to functions touched by this diff. Analyzed 0 functions across 0 files. (1 skipped: 1 unsupported.)

Full details: Security Anti-Patterns

Explanation

PASS. The commit changes only examples/diffusers/fastgen/requirements.txt; no modelopt or examples Python files changed. It replaces the existing nemo_automodel[diffusion] upper bound <1.0 with <0.6 and adds explanatory comments. It does not add a PIP dependency, and the changed content contains no prohibited deserialization, remote-code, eval/exec, or # nosec pattern.

✨ Finishing Touches

🧪 Generate unit tests (beta)

Create PR with unit tests
Commit unit tests in branch kmorabia/pin-nemo-automodel

_{Comment @coderabbitai help to get the list of available commands.}

codecov · 2026-08-26T23:23:42Z

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 78.92%. Comparing base (a0513f1) to head (b2924a0).

Additional details and impacted files

@@ Coverage Diff @@## main #2260 +/- ##
=======================================
Coverage 78.92% 78.92% =======================================
Files 523 523 Lines 60747 60747 =======================================
Hits 47943 47943 Misses 12804 12804

Flag	Coverage Δ
examples-diffusers	`20.69% <ø> (-0.02%)`	⬇️
examples-torch_onnx	`21.77% <ø> (-0.02%)`	⬇️
examples-torch_trt	`15.03% <ø> (-0.01%)`	⬇️
unit	`55.62% <ø> (ø)`

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Harness.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

github-actions · 2026-08-27T00:29:34Z

PR Preview Action v1.8.1
Preview removed because the pull request was closed.
2026-08-27 00:29 UTC

kevalmorabia97 · 2026-08-26T23:09:09Z

What does this PR do?

Type of change: Bug fix

nemo_automodel 0.6.0 removed nemo_automodel.recipes.diffusion.train.is_main_process without a replacement (it was a three-line rank-zero predicate in 0.5.0, and 0.6.0 defines no equivalent anywhere in the package). examples/diffusers/fastgen/dmd2_recipe.py imports it, so the example's import guard fires and every test in tests/examples/diffusers/ errors at collection:

ImportError: cannot import name 'is_main_process' from 'nemo_automodel.recipes.diffusion.train'
tests/examples/diffusers/fastgen/test_resume_dataloader.py
E ImportError: The DMD2 fastgen example requires `nemo_automodel`. ...
collected 42 items / 1 error

The requirement was >=0.4.0,<1.0, so CI picked 0.6.0 as soon as it was published and the onnx (diffusers) job started failing on every PR (e.g. runs 33020467654, 33019418460, 33010815298, 33007613265, 33006944292 — all unrelated branches). Capping at <0.6 restores the tested range.

Every other nemo_automodel symbol the example imports still exists in 0.6.0 (_diffusers.auto_diffusion_pipeline.NeMoAutoDiffusionPipeline, recipes.diffusion.train.TrainDiffusionRecipe, and the four components.datasets.diffusion.* helpers), so is_main_process is the only blocker; the alternative is defining that predicate locally and widening the cap again, which is worth doing separately if the example is meant to track 0.6.

Usage

pip install -r examples/diffusers/fastgen/requirements.txt

Testing

Reproduced the break by diffing the published wheels: is_main_process is defined at nemo_automodel/recipes/diffusion/train.py:692 in 0.5.0 and absent from 0.6.0 (grep -rn "def is_main_process" over the unpacked 0.6.0 wheel returns nothing). Confirmed the remaining imported symbols are all still present in 0.6.0.

CI on this PR exercises the fix directly: the onnx (diffusers) job installs from this requirements file and is the job that has been failing.

Before your PR is "Ready for review"

Is this change backward compatible?: ✅
If you copied code from any other sources or added a new PIP dependency, did you follow guidance in CONTRIBUTING.md: N/A — existing dependency, tightened bound.
Did you write any new necessary tests?: N/A — the existing tests/examples/diffusers/ suite is what this unblocks.
Did you update Changelog?: N/A — dependency-pin fix for a break introduced and fixed within the same unreleased cycle.
Did you get Claude approval on this PR?: ❌

🤖 Generated with Claude Code

Summary by CodeRabbit

Bug Fixes
- Fixed dependency compatibility for the FastGen diffusion example.
- Prevented installation of versions that could cause the example to fail at startup.

0.6.0 removed nemo_automodel.recipes.diffusion.train.is_main_process without a replacement, so dmd2_recipe.py fails to import and every diffusers example test errors at collection. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com> Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>

coderabbitai · 2026-08-26T23:09:59Z

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: c88c6187-1b4b-44f2-9939-04a9376fe97d

📥 Commits

Reviewing files that changed from the base of the PR and between a0513f1 and b2924a0.

📒 Files selected for processing (1)

examples/diffusers/fastgen/requirements.txt

Included review availability: Your plan provides up to 12 included reviews per hour; 7 remain after this review.

📝 Walkthrough

Walkthrough

The FastGen requirements file changes the nemo_automodel[diffusion] upper version bound from <1.0 to <0.6 while retaining the >=0.4.0 lower bound.

Changes

FastGen dependency compatibility

Layer / File(s)	Summary
Constrain diffusion dependency version `examples/diffusers/fastgen/requirements.txt`	The `nemo_automodel[diffusion]` requirement now allows versions from `0.4.0` up to, but not including, `0.6.0`.

Estimated code review effort: 1 (Trivial) | ~2 minutes

Merge Risk:⚪ Minimal · up to b2924

This localized dependency pin restores compatibility for the fastgen example, with no actionable merge-blocking risk remaining after normal checks and review.

Suggested reviewers:jingyu-ml, chenhanyu, shengliangxu

🚥 Pre-merge checks | ✅ 6

✅ Passed checks (6 passed)

Check name	Status	Explanation
Description Check	✅ Passed	Check skipped - CodeRabbit’s high-level summary is enabled.
Title check	✅ Passed	The title clearly and concisely describes the main change: pinning `nemo_automodel` below version 0.6 for the fastgen example.
Docstring Coverage	✅ Passed	No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check. Docstring coverage is scoped to functions touched by this diff. Analyzed 0 functions across 0…
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Security Anti-Patterns	✅ Passed	PASS. The commit changes only `examples/diffusers/fastgen/requirements.txt`; no `modelopt` or examples Python files changed. It replaces the existing `nemo_automodel[diffusion]` upper bound `<1.0` wit…

Full details: Docstring Coverage

Explanation

No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check. Docstring coverage is scoped to functions touched by this diff. Analyzed 0 functions across 0 files. (1 skipped: 1 unsupported.)

Full details: Security Anti-Patterns

Explanation

PASS. The commit changes only examples/diffusers/fastgen/requirements.txt; no modelopt or examples Python files changed. It replaces the existing nemo_automodel[diffusion] upper bound <1.0 with <0.6 and adds explanatory comments. It does not add a PIP dependency, and the changed content contains no prohibited deserialization, remote-code, eval/exec, or # nosec pattern.

✨ Finishing Touches

🧪 Generate unit tests (beta)

Create PR with unit tests
Commit unit tests in branch kmorabia/pin-nemo-automodel

_{Comment @coderabbitai help to get the list of available commands.}

codecov · 2026-08-26T23:23:42Z

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 78.92%. Comparing base (a0513f1) to head (b2924a0).

Additional details and impacted files

@@ Coverage Diff @@## main #2260 +/- ##
=======================================
Coverage 78.92% 78.92% =======================================
Files 523 523 Lines 60747 60747 =======================================
Hits 47943 47943 Misses 12804 12804

Flag	Coverage Δ
examples-diffusers	`20.69% <ø> (-0.02%)`	⬇️
examples-torch_onnx	`21.77% <ø> (-0.02%)`	⬇️
examples-torch_trt	`15.03% <ø> (-0.01%)`	⬇️
unit	`55.62% <ø> (ø)`

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Harness.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

github-actions · 2026-08-27T00:29:34Z

PR Preview Action v1.8.1
Preview removed because the pull request was closed.
2026-08-27 00:29 UTC

kevalmorabia97 · 2026-08-26T23:09:09Z

What does this PR do?

Type of change: Bug fix

nemo_automodel 0.6.0 removed nemo_automodel.recipes.diffusion.train.is_main_process without a replacement (it was a three-line rank-zero predicate in 0.5.0, and 0.6.0 defines no equivalent anywhere in the package). examples/diffusers/fastgen/dmd2_recipe.py imports it, so the example's import guard fires and every test in tests/examples/diffusers/ errors at collection:

ImportError: cannot import name 'is_main_process' from 'nemo_automodel.recipes.diffusion.train'
tests/examples/diffusers/fastgen/test_resume_dataloader.py
E ImportError: The DMD2 fastgen example requires `nemo_automodel`. ...
collected 42 items / 1 error

The requirement was >=0.4.0,<1.0, so CI picked 0.6.0 as soon as it was published and the onnx (diffusers) job started failing on every PR (e.g. runs 33020467654, 33019418460, 33010815298, 33007613265, 33006944292 — all unrelated branches). Capping at <0.6 restores the tested range.

Every other nemo_automodel symbol the example imports still exists in 0.6.0 (_diffusers.auto_diffusion_pipeline.NeMoAutoDiffusionPipeline, recipes.diffusion.train.TrainDiffusionRecipe, and the four components.datasets.diffusion.* helpers), so is_main_process is the only blocker; the alternative is defining that predicate locally and widening the cap again, which is worth doing separately if the example is meant to track 0.6.

Usage

pip install -r examples/diffusers/fastgen/requirements.txt

Testing

Reproduced the break by diffing the published wheels: is_main_process is defined at nemo_automodel/recipes/diffusion/train.py:692 in 0.5.0 and absent from 0.6.0 (grep -rn "def is_main_process" over the unpacked 0.6.0 wheel returns nothing). Confirmed the remaining imported symbols are all still present in 0.6.0.

CI on this PR exercises the fix directly: the onnx (diffusers) job installs from this requirements file and is the job that has been failing.

Before your PR is "Ready for review"

Is this change backward compatible?: ✅
If you copied code from any other sources or added a new PIP dependency, did you follow guidance in CONTRIBUTING.md: N/A — existing dependency, tightened bound.
Did you write any new necessary tests?: N/A — the existing tests/examples/diffusers/ suite is what this unblocks.
Did you update Changelog?: N/A — dependency-pin fix for a break introduced and fixed within the same unreleased cycle.
Did you get Claude approval on this PR?: ❌

🤖 Generated with Claude Code

Summary by CodeRabbit

Bug Fixes
- Fixed dependency compatibility for the FastGen diffusion example.
- Prevented installation of versions that could cause the example to fail at startup.

0.6.0 removed nemo_automodel.recipes.diffusion.train.is_main_process without a replacement, so dmd2_recipe.py fails to import and every diffusers example test errors at collection. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com> Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>

coderabbitai · 2026-08-26T23:09:59Z

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: c88c6187-1b4b-44f2-9939-04a9376fe97d

📥 Commits

Reviewing files that changed from the base of the PR and between a0513f1 and b2924a0.

📒 Files selected for processing (1)

examples/diffusers/fastgen/requirements.txt

Included review availability: Your plan provides up to 12 included reviews per hour; 7 remain after this review.

📝 Walkthrough

Walkthrough

The FastGen requirements file changes the nemo_automodel[diffusion] upper version bound from <1.0 to <0.6 while retaining the >=0.4.0 lower bound.

Changes

FastGen dependency compatibility

Layer / File(s)	Summary
Constrain diffusion dependency version `examples/diffusers/fastgen/requirements.txt`	The `nemo_automodel[diffusion]` requirement now allows versions from `0.4.0` up to, but not including, `0.6.0`.

Estimated code review effort: 1 (Trivial) | ~2 minutes

Merge Risk:⚪ Minimal · up to b2924

This localized dependency pin restores compatibility for the fastgen example, with no actionable merge-blocking risk remaining after normal checks and review.

Suggested reviewers:jingyu-ml, chenhanyu, shengliangxu

🚥 Pre-merge checks | ✅ 6

✅ Passed checks (6 passed)

Check name	Status	Explanation
Description Check	✅ Passed	Check skipped - CodeRabbit’s high-level summary is enabled.
Title check	✅ Passed	The title clearly and concisely describes the main change: pinning `nemo_automodel` below version 0.6 for the fastgen example.
Docstring Coverage	✅ Passed	No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check. Docstring coverage is scoped to functions touched by this diff. Analyzed 0 functions across 0…
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Security Anti-Patterns	✅ Passed	PASS. The commit changes only `examples/diffusers/fastgen/requirements.txt`; no `modelopt` or examples Python files changed. It replaces the existing `nemo_automodel[diffusion]` upper bound `<1.0` wit…

Full details: Docstring Coverage

Explanation

No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check. Docstring coverage is scoped to functions touched by this diff. Analyzed 0 functions across 0 files. (1 skipped: 1 unsupported.)

Full details: Security Anti-Patterns

Explanation

PASS. The commit changes only examples/diffusers/fastgen/requirements.txt; no modelopt or examples Python files changed. It replaces the existing nemo_automodel[diffusion] upper bound <1.0 with <0.6 and adds explanatory comments. It does not add a PIP dependency, and the changed content contains no prohibited deserialization, remote-code, eval/exec, or # nosec pattern.

✨ Finishing Touches

🧪 Generate unit tests (beta)

Create PR with unit tests
Commit unit tests in branch kmorabia/pin-nemo-automodel

_{Comment @coderabbitai help to get the list of available commands.}

codecov · 2026-08-26T23:23:42Z

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 78.92%. Comparing base (a0513f1) to head (b2924a0).

Additional details and impacted files

@@ Coverage Diff @@## main #2260 +/- ##
=======================================
Coverage 78.92% 78.92% =======================================
Files 523 523 Lines 60747 60747 =======================================
Hits 47943 47943 Misses 12804 12804

Flag	Coverage Δ
examples-diffusers	`20.69% <ø> (-0.02%)`	⬇️
examples-torch_onnx	`21.77% <ø> (-0.02%)`	⬇️
examples-torch_trt	`15.03% <ø> (-0.01%)`	⬇️
unit	`55.62% <ø> (ø)`

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Harness.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

github-actions · 2026-08-27T00:29:34Z

PR Preview Action v1.8.1
Preview removed because the pull request was closed.
2026-08-27 00:29 UTC

kevalmorabia97 · 2026-08-26T23:09:09Z

What does this PR do?

Type of change: Bug fix

nemo_automodel 0.6.0 removed nemo_automodel.recipes.diffusion.train.is_main_process without a replacement (it was a three-line rank-zero predicate in 0.5.0, and 0.6.0 defines no equivalent anywhere in the package). examples/diffusers/fastgen/dmd2_recipe.py imports it, so the example's import guard fires and every test in tests/examples/diffusers/ errors at collection:

ImportError: cannot import name 'is_main_process' from 'nemo_automodel.recipes.diffusion.train'
tests/examples/diffusers/fastgen/test_resume_dataloader.py
E ImportError: The DMD2 fastgen example requires `nemo_automodel`. ...
collected 42 items / 1 error

The requirement was >=0.4.0,<1.0, so CI picked 0.6.0 as soon as it was published and the onnx (diffusers) job started failing on every PR (e.g. runs 33020467654, 33019418460, 33010815298, 33007613265, 33006944292 — all unrelated branches). Capping at <0.6 restores the tested range.

Every other nemo_automodel symbol the example imports still exists in 0.6.0 (_diffusers.auto_diffusion_pipeline.NeMoAutoDiffusionPipeline, recipes.diffusion.train.TrainDiffusionRecipe, and the four components.datasets.diffusion.* helpers), so is_main_process is the only blocker; the alternative is defining that predicate locally and widening the cap again, which is worth doing separately if the example is meant to track 0.6.

Usage

pip install -r examples/diffusers/fastgen/requirements.txt

Testing

Reproduced the break by diffing the published wheels: is_main_process is defined at nemo_automodel/recipes/diffusion/train.py:692 in 0.5.0 and absent from 0.6.0 (grep -rn "def is_main_process" over the unpacked 0.6.0 wheel returns nothing). Confirmed the remaining imported symbols are all still present in 0.6.0.

CI on this PR exercises the fix directly: the onnx (diffusers) job installs from this requirements file and is the job that has been failing.

Before your PR is "Ready for review"

Is this change backward compatible?: ✅
If you copied code from any other sources or added a new PIP dependency, did you follow guidance in CONTRIBUTING.md: N/A — existing dependency, tightened bound.
Did you write any new necessary tests?: N/A — the existing tests/examples/diffusers/ suite is what this unblocks.
Did you update Changelog?: N/A — dependency-pin fix for a break introduced and fixed within the same unreleased cycle.
Did you get Claude approval on this PR?: ❌

🤖 Generated with Claude Code

Summary by CodeRabbit

Bug Fixes
- Fixed dependency compatibility for the FastGen diffusion example.
- Prevented installation of versions that could cause the example to fail at startup.

0.6.0 removed nemo_automodel.recipes.diffusion.train.is_main_process without a replacement, so dmd2_recipe.py fails to import and every diffusers example test errors at collection. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com> Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>

coderabbitai · 2026-08-26T23:09:59Z

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: c88c6187-1b4b-44f2-9939-04a9376fe97d

📥 Commits

Reviewing files that changed from the base of the PR and between a0513f1 and b2924a0.

📒 Files selected for processing (1)

examples/diffusers/fastgen/requirements.txt

Included review availability: Your plan provides up to 12 included reviews per hour; 7 remain after this review.

📝 Walkthrough

Walkthrough

The FastGen requirements file changes the nemo_automodel[diffusion] upper version bound from <1.0 to <0.6 while retaining the >=0.4.0 lower bound.

Changes

FastGen dependency compatibility

Layer / File(s)	Summary
Constrain diffusion dependency version `examples/diffusers/fastgen/requirements.txt`	The `nemo_automodel[diffusion]` requirement now allows versions from `0.4.0` up to, but not including, `0.6.0`.

Estimated code review effort: 1 (Trivial) | ~2 minutes

Merge Risk:⚪ Minimal · up to b2924

This localized dependency pin restores compatibility for the fastgen example, with no actionable merge-blocking risk remaining after normal checks and review.

Suggested reviewers:jingyu-ml, chenhanyu, shengliangxu

🚥 Pre-merge checks | ✅ 6

✅ Passed checks (6 passed)

Check name	Status	Explanation
Description Check	✅ Passed	Check skipped - CodeRabbit’s high-level summary is enabled.
Title check	✅ Passed	The title clearly and concisely describes the main change: pinning `nemo_automodel` below version 0.6 for the fastgen example.
Docstring Coverage	✅ Passed	No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check. Docstring coverage is scoped to functions touched by this diff. Analyzed 0 functions across 0…
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Security Anti-Patterns	✅ Passed	PASS. The commit changes only `examples/diffusers/fastgen/requirements.txt`; no `modelopt` or examples Python files changed. It replaces the existing `nemo_automodel[diffusion]` upper bound `<1.0` wit…

Full details: Docstring Coverage

Explanation

No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check. Docstring coverage is scoped to functions touched by this diff. Analyzed 0 functions across 0 files. (1 skipped: 1 unsupported.)

Full details: Security Anti-Patterns

Explanation

PASS. The commit changes only examples/diffusers/fastgen/requirements.txt; no modelopt or examples Python files changed. It replaces the existing nemo_automodel[diffusion] upper bound <1.0 with <0.6 and adds explanatory comments. It does not add a PIP dependency, and the changed content contains no prohibited deserialization, remote-code, eval/exec, or # nosec pattern.

✨ Finishing Touches

🧪 Generate unit tests (beta)

Create PR with unit tests
Commit unit tests in branch kmorabia/pin-nemo-automodel

_{Comment @coderabbitai help to get the list of available commands.}

codecov · 2026-08-26T23:23:42Z

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 78.92%. Comparing base (a0513f1) to head (b2924a0).

Additional details and impacted files

@@ Coverage Diff @@## main #2260 +/- ##
=======================================
Coverage 78.92% 78.92% =======================================
Files 523 523 Lines 60747 60747 =======================================
Hits 47943 47943 Misses 12804 12804

Flag	Coverage Δ
examples-diffusers	`20.69% <ø> (-0.02%)`	⬇️
examples-torch_onnx	`21.77% <ø> (-0.02%)`	⬇️
examples-torch_trt	`15.03% <ø> (-0.01%)`	⬇️
unit	`55.62% <ø> (ø)`

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Harness.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

github-actions · 2026-08-27T00:29:34Z

PR Preview Action v1.8.1
Preview removed because the pull request was closed.
2026-08-27 00:29 UTC

kevalmorabia97 · 2026-08-26T23:09:09Z

What does this PR do?

Type of change: Bug fix

nemo_automodel 0.6.0 removed nemo_automodel.recipes.diffusion.train.is_main_process without a replacement (it was a three-line rank-zero predicate in 0.5.0, and 0.6.0 defines no equivalent anywhere in the package). examples/diffusers/fastgen/dmd2_recipe.py imports it, so the example's import guard fires and every test in tests/examples/diffusers/ errors at collection:

ImportError: cannot import name 'is_main_process' from 'nemo_automodel.recipes.diffusion.train'
tests/examples/diffusers/fastgen/test_resume_dataloader.py
E ImportError: The DMD2 fastgen example requires `nemo_automodel`. ...
collected 42 items / 1 error

The requirement was >=0.4.0,<1.0, so CI picked 0.6.0 as soon as it was published and the onnx (diffusers) job started failing on every PR (e.g. runs 33020467654, 33019418460, 33010815298, 33007613265, 33006944292 — all unrelated branches). Capping at <0.6 restores the tested range.

Every other nemo_automodel symbol the example imports still exists in 0.6.0 (_diffusers.auto_diffusion_pipeline.NeMoAutoDiffusionPipeline, recipes.diffusion.train.TrainDiffusionRecipe, and the four components.datasets.diffusion.* helpers), so is_main_process is the only blocker; the alternative is defining that predicate locally and widening the cap again, which is worth doing separately if the example is meant to track 0.6.

Usage

pip install -r examples/diffusers/fastgen/requirements.txt

Testing

Reproduced the break by diffing the published wheels: is_main_process is defined at nemo_automodel/recipes/diffusion/train.py:692 in 0.5.0 and absent from 0.6.0 (grep -rn "def is_main_process" over the unpacked 0.6.0 wheel returns nothing). Confirmed the remaining imported symbols are all still present in 0.6.0.

CI on this PR exercises the fix directly: the onnx (diffusers) job installs from this requirements file and is the job that has been failing.

Before your PR is "Ready for review"

Is this change backward compatible?: ✅
If you copied code from any other sources or added a new PIP dependency, did you follow guidance in CONTRIBUTING.md: N/A — existing dependency, tightened bound.
Did you write any new necessary tests?: N/A — the existing tests/examples/diffusers/ suite is what this unblocks.
Did you update Changelog?: N/A — dependency-pin fix for a break introduced and fixed within the same unreleased cycle.
Did you get Claude approval on this PR?: ❌

🤖 Generated with Claude Code

Summary by CodeRabbit

Bug Fixes
- Fixed dependency compatibility for the FastGen diffusion example.
- Prevented installation of versions that could cause the example to fail at startup.

0.6.0 removed nemo_automodel.recipes.diffusion.train.is_main_process without a replacement, so dmd2_recipe.py fails to import and every diffusers example test errors at collection. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com> Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>

coderabbitai · 2026-08-26T23:09:59Z

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: c88c6187-1b4b-44f2-9939-04a9376fe97d

📥 Commits

Reviewing files that changed from the base of the PR and between a0513f1 and b2924a0.

📒 Files selected for processing (1)

examples/diffusers/fastgen/requirements.txt

Included review availability: Your plan provides up to 12 included reviews per hour; 7 remain after this review.

📝 Walkthrough

Walkthrough

The FastGen requirements file changes the nemo_automodel[diffusion] upper version bound from <1.0 to <0.6 while retaining the >=0.4.0 lower bound.

Changes

FastGen dependency compatibility

Layer / File(s)	Summary
Constrain diffusion dependency version `examples/diffusers/fastgen/requirements.txt`	The `nemo_automodel[diffusion]` requirement now allows versions from `0.4.0` up to, but not including, `0.6.0`.

Estimated code review effort: 1 (Trivial) | ~2 minutes

Merge Risk:⚪ Minimal · up to b2924

This localized dependency pin restores compatibility for the fastgen example, with no actionable merge-blocking risk remaining after normal checks and review.

Suggested reviewers:jingyu-ml, chenhanyu, shengliangxu

🚥 Pre-merge checks | ✅ 6

✅ Passed checks (6 passed)

Check name	Status	Explanation
Description Check	✅ Passed	Check skipped - CodeRabbit’s high-level summary is enabled.
Title check	✅ Passed	The title clearly and concisely describes the main change: pinning `nemo_automodel` below version 0.6 for the fastgen example.
Docstring Coverage	✅ Passed	No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check. Docstring coverage is scoped to functions touched by this diff. Analyzed 0 functions across 0…
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Security Anti-Patterns	✅ Passed	PASS. The commit changes only `examples/diffusers/fastgen/requirements.txt`; no `modelopt` or examples Python files changed. It replaces the existing `nemo_automodel[diffusion]` upper bound `<1.0` wit…

Full details: Docstring Coverage

Explanation

No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check. Docstring coverage is scoped to functions touched by this diff. Analyzed 0 functions across 0 files. (1 skipped: 1 unsupported.)

Full details: Security Anti-Patterns

Explanation

PASS. The commit changes only examples/diffusers/fastgen/requirements.txt; no modelopt or examples Python files changed. It replaces the existing nemo_automodel[diffusion] upper bound <1.0 with <0.6 and adds explanatory comments. It does not add a PIP dependency, and the changed content contains no prohibited deserialization, remote-code, eval/exec, or # nosec pattern.

✨ Finishing Touches

🧪 Generate unit tests (beta)

Create PR with unit tests
Commit unit tests in branch kmorabia/pin-nemo-automodel

_{Comment @coderabbitai help to get the list of available commands.}

codecov · 2026-08-26T23:23:42Z

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 78.92%. Comparing base (a0513f1) to head (b2924a0).

Additional details and impacted files

@@ Coverage Diff @@## main #2260 +/- ##
=======================================
Coverage 78.92% 78.92% =======================================
Files 523 523 Lines 60747 60747 =======================================
Hits 47943 47943 Misses 12804 12804

Flag	Coverage Δ
examples-diffusers	`20.69% <ø> (-0.02%)`	⬇️
examples-torch_onnx	`21.77% <ø> (-0.02%)`	⬇️
examples-torch_trt	`15.03% <ø> (-0.01%)`	⬇️
unit	`55.62% <ø> (ø)`

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Harness.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

github-actions · 2026-08-27T00:29:34Z

PR Preview Action v1.8.1
Preview removed because the pull request was closed.
2026-08-27 00:29 UTC

kevalmorabia97 · 2026-08-26T23:09:09Z

What does this PR do?

Type of change: Bug fix

nemo_automodel 0.6.0 removed nemo_automodel.recipes.diffusion.train.is_main_process without a replacement (it was a three-line rank-zero predicate in 0.5.0, and 0.6.0 defines no equivalent anywhere in the package). examples/diffusers/fastgen/dmd2_recipe.py imports it, so the example's import guard fires and every test in tests/examples/diffusers/ errors at collection:

ImportError: cannot import name 'is_main_process' from 'nemo_automodel.recipes.diffusion.train'
tests/examples/diffusers/fastgen/test_resume_dataloader.py
E ImportError: The DMD2 fastgen example requires `nemo_automodel`. ...
collected 42 items / 1 error

The requirement was >=0.4.0,<1.0, so CI picked 0.6.0 as soon as it was published and the onnx (diffusers) job started failing on every PR (e.g. runs 33020467654, 33019418460, 33010815298, 33007613265, 33006944292 — all unrelated branches). Capping at <0.6 restores the tested range.

Every other nemo_automodel symbol the example imports still exists in 0.6.0 (_diffusers.auto_diffusion_pipeline.NeMoAutoDiffusionPipeline, recipes.diffusion.train.TrainDiffusionRecipe, and the four components.datasets.diffusion.* helpers), so is_main_process is the only blocker; the alternative is defining that predicate locally and widening the cap again, which is worth doing separately if the example is meant to track 0.6.

Usage

pip install -r examples/diffusers/fastgen/requirements.txt

Testing

Reproduced the break by diffing the published wheels: is_main_process is defined at nemo_automodel/recipes/diffusion/train.py:692 in 0.5.0 and absent from 0.6.0 (grep -rn "def is_main_process" over the unpacked 0.6.0 wheel returns nothing). Confirmed the remaining imported symbols are all still present in 0.6.0.

CI on this PR exercises the fix directly: the onnx (diffusers) job installs from this requirements file and is the job that has been failing.

Before your PR is "Ready for review"

Is this change backward compatible?: ✅
If you copied code from any other sources or added a new PIP dependency, did you follow guidance in CONTRIBUTING.md: N/A — existing dependency, tightened bound.
Did you write any new necessary tests?: N/A — the existing tests/examples/diffusers/ suite is what this unblocks.
Did you update Changelog?: N/A — dependency-pin fix for a break introduced and fixed within the same unreleased cycle.
Did you get Claude approval on this PR?: ❌

🤖 Generated with Claude Code

Summary by CodeRabbit

Bug Fixes
- Fixed dependency compatibility for the FastGen diffusion example.
- Prevented installation of versions that could cause the example to fail at startup.

0.6.0 removed nemo_automodel.recipes.diffusion.train.is_main_process without a replacement, so dmd2_recipe.py fails to import and every diffusers example test errors at collection. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com> Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>

coderabbitai · 2026-08-26T23:09:59Z

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: c88c6187-1b4b-44f2-9939-04a9376fe97d

📥 Commits

Reviewing files that changed from the base of the PR and between a0513f1 and b2924a0.

📒 Files selected for processing (1)

examples/diffusers/fastgen/requirements.txt

Included review availability: Your plan provides up to 12 included reviews per hour; 7 remain after this review.

📝 Walkthrough

Walkthrough

The FastGen requirements file changes the nemo_automodel[diffusion] upper version bound from <1.0 to <0.6 while retaining the >=0.4.0 lower bound.

Changes

FastGen dependency compatibility

Layer / File(s)	Summary
Constrain diffusion dependency version `examples/diffusers/fastgen/requirements.txt`	The `nemo_automodel[diffusion]` requirement now allows versions from `0.4.0` up to, but not including, `0.6.0`.

Estimated code review effort: 1 (Trivial) | ~2 minutes

Merge Risk:⚪ Minimal · up to b2924

This localized dependency pin restores compatibility for the fastgen example, with no actionable merge-blocking risk remaining after normal checks and review.

Suggested reviewers:jingyu-ml, chenhanyu, shengliangxu

🚥 Pre-merge checks | ✅ 6

✅ Passed checks (6 passed)

Check name	Status	Explanation
Description Check	✅ Passed	Check skipped - CodeRabbit’s high-level summary is enabled.
Title check	✅ Passed	The title clearly and concisely describes the main change: pinning `nemo_automodel` below version 0.6 for the fastgen example.
Docstring Coverage	✅ Passed	No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check. Docstring coverage is scoped to functions touched by this diff. Analyzed 0 functions across 0…
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Security Anti-Patterns	✅ Passed	PASS. The commit changes only `examples/diffusers/fastgen/requirements.txt`; no `modelopt` or examples Python files changed. It replaces the existing `nemo_automodel[diffusion]` upper bound `<1.0` wit…

Full details: Docstring Coverage

Explanation

No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check. Docstring coverage is scoped to functions touched by this diff. Analyzed 0 functions across 0 files. (1 skipped: 1 unsupported.)

Full details: Security Anti-Patterns

Explanation

PASS. The commit changes only examples/diffusers/fastgen/requirements.txt; no modelopt or examples Python files changed. It replaces the existing nemo_automodel[diffusion] upper bound <1.0 with <0.6 and adds explanatory comments. It does not add a PIP dependency, and the changed content contains no prohibited deserialization, remote-code, eval/exec, or # nosec pattern.

✨ Finishing Touches

🧪 Generate unit tests (beta)

Create PR with unit tests
Commit unit tests in branch kmorabia/pin-nemo-automodel

_{Comment @coderabbitai help to get the list of available commands.}

codecov · 2026-08-26T23:23:42Z

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 78.92%. Comparing base (a0513f1) to head (b2924a0).

Additional details and impacted files

@@ Coverage Diff @@## main #2260 +/- ##
=======================================
Coverage 78.92% 78.92% =======================================
Files 523 523 Lines 60747 60747 =======================================
Hits 47943 47943 Misses 12804 12804

Flag	Coverage Δ
examples-diffusers	`20.69% <ø> (-0.02%)`	⬇️
examples-torch_onnx	`21.77% <ø> (-0.02%)`	⬇️
examples-torch_trt	`15.03% <ø> (-0.01%)`	⬇️
unit	`55.62% <ø> (ø)`

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Harness.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

github-actions · 2026-08-27T00:29:34Z

PR Preview Action v1.8.1
Preview removed because the pull request was closed.
2026-08-27 00:29 UTC

kevalmorabia97 · 2026-08-26T23:09:09Z

What does this PR do?

Type of change: Bug fix

nemo_automodel 0.6.0 removed nemo_automodel.recipes.diffusion.train.is_main_process without a replacement (it was a three-line rank-zero predicate in 0.5.0, and 0.6.0 defines no equivalent anywhere in the package). examples/diffusers/fastgen/dmd2_recipe.py imports it, so the example's import guard fires and every test in tests/examples/diffusers/ errors at collection:

ImportError: cannot import name 'is_main_process' from 'nemo_automodel.recipes.diffusion.train'
tests/examples/diffusers/fastgen/test_resume_dataloader.py
E ImportError: The DMD2 fastgen example requires `nemo_automodel`. ...
collected 42 items / 1 error

The requirement was >=0.4.0,<1.0, so CI picked 0.6.0 as soon as it was published and the onnx (diffusers) job started failing on every PR (e.g. runs 33020467654, 33019418460, 33010815298, 33007613265, 33006944292 — all unrelated branches). Capping at <0.6 restores the tested range.

Every other nemo_automodel symbol the example imports still exists in 0.6.0 (_diffusers.auto_diffusion_pipeline.NeMoAutoDiffusionPipeline, recipes.diffusion.train.TrainDiffusionRecipe, and the four components.datasets.diffusion.* helpers), so is_main_process is the only blocker; the alternative is defining that predicate locally and widening the cap again, which is worth doing separately if the example is meant to track 0.6.

Usage

pip install -r examples/diffusers/fastgen/requirements.txt

Testing

Reproduced the break by diffing the published wheels: is_main_process is defined at nemo_automodel/recipes/diffusion/train.py:692 in 0.5.0 and absent from 0.6.0 (grep -rn "def is_main_process" over the unpacked 0.6.0 wheel returns nothing). Confirmed the remaining imported symbols are all still present in 0.6.0.

CI on this PR exercises the fix directly: the onnx (diffusers) job installs from this requirements file and is the job that has been failing.

Before your PR is "Ready for review"

Is this change backward compatible?: ✅
If you copied code from any other sources or added a new PIP dependency, did you follow guidance in CONTRIBUTING.md: N/A — existing dependency, tightened bound.
Did you write any new necessary tests?: N/A — the existing tests/examples/diffusers/ suite is what this unblocks.
Did you update Changelog?: N/A — dependency-pin fix for a break introduced and fixed within the same unreleased cycle.
Did you get Claude approval on this PR?: ❌

🤖 Generated with Claude Code

Summary by CodeRabbit

Bug Fixes
- Fixed dependency compatibility for the FastGen diffusion example.
- Prevented installation of versions that could cause the example to fail at startup.

0.6.0 removed nemo_automodel.recipes.diffusion.train.is_main_process without a replacement, so dmd2_recipe.py fails to import and every diffusers example test errors at collection. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com> Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>

coderabbitai · 2026-08-26T23:09:59Z

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: c88c6187-1b4b-44f2-9939-04a9376fe97d

📥 Commits

Reviewing files that changed from the base of the PR and between a0513f1 and b2924a0.

📒 Files selected for processing (1)

examples/diffusers/fastgen/requirements.txt

Included review availability: Your plan provides up to 12 included reviews per hour; 7 remain after this review.

📝 Walkthrough

Walkthrough

The FastGen requirements file changes the nemo_automodel[diffusion] upper version bound from <1.0 to <0.6 while retaining the >=0.4.0 lower bound.

Changes

FastGen dependency compatibility

Layer / File(s)	Summary
Constrain diffusion dependency version `examples/diffusers/fastgen/requirements.txt`	The `nemo_automodel[diffusion]` requirement now allows versions from `0.4.0` up to, but not including, `0.6.0`.

Estimated code review effort: 1 (Trivial) | ~2 minutes

Merge Risk:⚪ Minimal · up to b2924

This localized dependency pin restores compatibility for the fastgen example, with no actionable merge-blocking risk remaining after normal checks and review.

Suggested reviewers:jingyu-ml, chenhanyu, shengliangxu

🚥 Pre-merge checks | ✅ 6

✅ Passed checks (6 passed)

Check name	Status	Explanation
Description Check	✅ Passed	Check skipped - CodeRabbit’s high-level summary is enabled.
Title check	✅ Passed	The title clearly and concisely describes the main change: pinning `nemo_automodel` below version 0.6 for the fastgen example.
Docstring Coverage	✅ Passed	No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check. Docstring coverage is scoped to functions touched by this diff. Analyzed 0 functions across 0…
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Security Anti-Patterns	✅ Passed	PASS. The commit changes only `examples/diffusers/fastgen/requirements.txt`; no `modelopt` or examples Python files changed. It replaces the existing `nemo_automodel[diffusion]` upper bound `<1.0` wit…

Full details: Docstring Coverage

Explanation

No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check. Docstring coverage is scoped to functions touched by this diff. Analyzed 0 functions across 0 files. (1 skipped: 1 unsupported.)

Full details: Security Anti-Patterns

Explanation

PASS. The commit changes only examples/diffusers/fastgen/requirements.txt; no modelopt or examples Python files changed. It replaces the existing nemo_automodel[diffusion] upper bound <1.0 with <0.6 and adds explanatory comments. It does not add a PIP dependency, and the changed content contains no prohibited deserialization, remote-code, eval/exec, or # nosec pattern.

✨ Finishing Touches

🧪 Generate unit tests (beta)

Create PR with unit tests
Commit unit tests in branch kmorabia/pin-nemo-automodel

_{Comment @coderabbitai help to get the list of available commands.}

codecov · 2026-08-26T23:23:42Z

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 78.92%. Comparing base (a0513f1) to head (b2924a0).

Additional details and impacted files

@@ Coverage Diff @@## main #2260 +/- ##
=======================================
Coverage 78.92% 78.92% =======================================
Files 523 523 Lines 60747 60747 =======================================
Hits 47943 47943 Misses 12804 12804

Flag	Coverage Δ
examples-diffusers	`20.69% <ø> (-0.02%)`	⬇️
examples-torch_onnx	`21.77% <ø> (-0.02%)`	⬇️
examples-torch_trt	`15.03% <ø> (-0.01%)`	⬇️
unit	`55.62% <ø> (ø)`

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Harness.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

github-actions · 2026-08-27T00:29:34Z

PR Preview Action v1.8.1
Preview removed because the pull request was closed.
2026-08-27 00:29 UTC

kevalmorabia97 · 2026-08-26T23:09:09Z

What does this PR do?

Type of change: Bug fix

nemo_automodel 0.6.0 removed nemo_automodel.recipes.diffusion.train.is_main_process without a replacement (it was a three-line rank-zero predicate in 0.5.0, and 0.6.0 defines no equivalent anywhere in the package). examples/diffusers/fastgen/dmd2_recipe.py imports it, so the example's import guard fires and every test in tests/examples/diffusers/ errors at collection:

ImportError: cannot import name 'is_main_process' from 'nemo_automodel.recipes.diffusion.train'
tests/examples/diffusers/fastgen/test_resume_dataloader.py
E ImportError: The DMD2 fastgen example requires `nemo_automodel`. ...
collected 42 items / 1 error

The requirement was >=0.4.0,<1.0, so CI picked 0.6.0 as soon as it was published and the onnx (diffusers) job started failing on every PR (e.g. runs 33020467654, 33019418460, 33010815298, 33007613265, 33006944292 — all unrelated branches). Capping at <0.6 restores the tested range.

Every other nemo_automodel symbol the example imports still exists in 0.6.0 (_diffusers.auto_diffusion_pipeline.NeMoAutoDiffusionPipeline, recipes.diffusion.train.TrainDiffusionRecipe, and the four components.datasets.diffusion.* helpers), so is_main_process is the only blocker; the alternative is defining that predicate locally and widening the cap again, which is worth doing separately if the example is meant to track 0.6.

Usage

pip install -r examples/diffusers/fastgen/requirements.txt

Testing

Reproduced the break by diffing the published wheels: is_main_process is defined at nemo_automodel/recipes/diffusion/train.py:692 in 0.5.0 and absent from 0.6.0 (grep -rn "def is_main_process" over the unpacked 0.6.0 wheel returns nothing). Confirmed the remaining imported symbols are all still present in 0.6.0.

CI on this PR exercises the fix directly: the onnx (diffusers) job installs from this requirements file and is the job that has been failing.

Before your PR is "Ready for review"

Is this change backward compatible?: ✅
If you copied code from any other sources or added a new PIP dependency, did you follow guidance in CONTRIBUTING.md: N/A — existing dependency, tightened bound.
Did you write any new necessary tests?: N/A — the existing tests/examples/diffusers/ suite is what this unblocks.
Did you update Changelog?: N/A — dependency-pin fix for a break introduced and fixed within the same unreleased cycle.
Did you get Claude approval on this PR?: ❌

🤖 Generated with Claude Code

Summary by CodeRabbit

Bug Fixes
- Fixed dependency compatibility for the FastGen diffusion example.
- Prevented installation of versions that could cause the example to fail at startup.

0.6.0 removed nemo_automodel.recipes.diffusion.train.is_main_process without a replacement, so dmd2_recipe.py fails to import and every diffusers example test errors at collection. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com> Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>

coderabbitai · 2026-08-26T23:09:59Z

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: c88c6187-1b4b-44f2-9939-04a9376fe97d

📥 Commits

Reviewing files that changed from the base of the PR and between a0513f1 and b2924a0.

📒 Files selected for processing (1)

examples/diffusers/fastgen/requirements.txt

Included review availability: Your plan provides up to 12 included reviews per hour; 7 remain after this review.

📝 Walkthrough

Walkthrough

The FastGen requirements file changes the nemo_automodel[diffusion] upper version bound from <1.0 to <0.6 while retaining the >=0.4.0 lower bound.

Changes

FastGen dependency compatibility

Layer / File(s)	Summary
Constrain diffusion dependency version `examples/diffusers/fastgen/requirements.txt`	The `nemo_automodel[diffusion]` requirement now allows versions from `0.4.0` up to, but not including, `0.6.0`.

Estimated code review effort: 1 (Trivial) | ~2 minutes

Merge Risk:⚪ Minimal · up to b2924

This localized dependency pin restores compatibility for the fastgen example, with no actionable merge-blocking risk remaining after normal checks and review.

Suggested reviewers:jingyu-ml, chenhanyu, shengliangxu

🚥 Pre-merge checks | ✅ 6

✅ Passed checks (6 passed)

Check name	Status	Explanation
Description Check	✅ Passed	Check skipped - CodeRabbit’s high-level summary is enabled.
Title check	✅ Passed	The title clearly and concisely describes the main change: pinning `nemo_automodel` below version 0.6 for the fastgen example.
Docstring Coverage	✅ Passed	No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check. Docstring coverage is scoped to functions touched by this diff. Analyzed 0 functions across 0…
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Security Anti-Patterns	✅ Passed	PASS. The commit changes only `examples/diffusers/fastgen/requirements.txt`; no `modelopt` or examples Python files changed. It replaces the existing `nemo_automodel[diffusion]` upper bound `<1.0` wit…

Full details: Docstring Coverage

Explanation

No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check. Docstring coverage is scoped to functions touched by this diff. Analyzed 0 functions across 0 files. (1 skipped: 1 unsupported.)

Full details: Security Anti-Patterns

Explanation

PASS. The commit changes only examples/diffusers/fastgen/requirements.txt; no modelopt or examples Python files changed. It replaces the existing nemo_automodel[diffusion] upper bound <1.0 with <0.6 and adds explanatory comments. It does not add a PIP dependency, and the changed content contains no prohibited deserialization, remote-code, eval/exec, or # nosec pattern.

✨ Finishing Touches

🧪 Generate unit tests (beta)

Create PR with unit tests
Commit unit tests in branch kmorabia/pin-nemo-automodel

_{Comment @coderabbitai help to get the list of available commands.}

codecov · 2026-08-26T23:23:42Z

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 78.92%. Comparing base (a0513f1) to head (b2924a0).

Additional details and impacted files

@@ Coverage Diff @@## main #2260 +/- ##
=======================================
Coverage 78.92% 78.92% =======================================
Files 523 523 Lines 60747 60747 =======================================
Hits 47943 47943 Misses 12804 12804

Flag	Coverage Δ
examples-diffusers	`20.69% <ø> (-0.02%)`	⬇️
examples-torch_onnx	`21.77% <ø> (-0.02%)`	⬇️
examples-torch_trt	`15.03% <ø> (-0.01%)`	⬇️
unit	`55.62% <ø> (ø)`

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Harness.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

github-actions · 2026-08-27T00:29:34Z

PR Preview Action v1.8.1
Preview removed because the pull request was closed.
2026-08-27 00:29 UTC

kevalmorabia97 requested a review from a team as a code owner August 26, 2026 23:09

kevalmorabia97 requested a review from jingyu-ml August 26, 2026 23:09

coderabbitaiBot approved these changes Aug 26, 2026
View reviewed changes

kevalmorabia97 merged commit 449a399 into mainAug 27, 2026
34 checks passed

kevalmorabia97 deleted the kmorabia/pin-nemo-automodel branch August 27, 2026 00:29

yeyu-nvidia mentioned this pull request Sep 3, 2026
Consolidate speculative-decoding agent skills into one stage/algorithm tree #2201
Open