Skip to content

ci: Move source install fla to dev group#1580

Merged
thomasdhc merged 3 commits intomainfrom
donghyukc/fla_install_fix
Mar 20, 2026
Merged

ci: Move source install fla to dev group#1580
thomasdhc merged 3 commits intomainfrom
donghyukc/fla_install_fix

Conversation

@thomasdhc
Copy link
Copy Markdown
Contributor

What does this PR do ?

Add a one line overview of what this PR aims to accomplish.

Changelog

  • Add specific line by line info of high level changes in this PR.

Before your PR is "Ready for review"

Pre checks:

  • Make sure you read and followed Contributor guidelines
  • Did you write any new necessary tests?
  • Did you add or update any necessary documentation?

If you haven't finished some of the above items you can still open "Draft" PR.

Additional Information

  • Related to # (issue)

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>
@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot Bot commented Mar 19, 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.

@thomasdhc
Copy link
Copy Markdown
Contributor Author

/ok to test 822c773

@thomasdhc thomasdhc enabled auto-merge (squash) March 19, 2026 23:57
@thomasdhc thomasdhc merged commit 2297b1a into main Mar 20, 2026
53 checks passed
@thomasdhc thomasdhc deleted the donghyukc/fla_install_fix branch March 20, 2026 00:33
pthombre pushed a commit that referenced this pull request Mar 20, 2026
* Move source install fla to dev group

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>

* Update uv lock

Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>

---------

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>
Co-authored-by: NeMo Bot <nemo-bot@nvidia.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
akoumpa added a commit that referenced this pull request Mar 20, 2026
* feat: Integrate Wan with multi-resolution DL

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Required changes for compatability with AM container

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Fix overrides

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* feat: Add docs about diffusion support in AM

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Remove older data processing tools

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/diffusion/dataset.md

Co-authored-by: Andrew Chen <chenopis@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Apply suggestions from code review

Co-authored-by: Andrew Chen <chenopis@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Revert non-doc code changes to match main

This PR is docs-only; restore test and tool files to main's state.

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Restructure the diffusion finetuning doc

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* fix: Nemotron v3 inputs_embeds generation (#1583)

Fix Nemotron v3 inputs_embeds generation

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* fix: checkpointing for PEFT. (#1576)

* Fix checkpointing for PEFT.

Previously, the state_dict in the modelstate class had an if/elseif/else statement where
peft was handled in two caces and non-peft on the third one.

The first case of peft, was handling correctly, while the second was including
buffers causing issues in downstream consumers.

This fix simplifies the logic (simple if/else) and bypassed the issues with the
buffer.

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* Update stateful_wrappers.py

* fix

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* update test

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* improve error logging in test; pass is_peft to optimizerstate

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* fix

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* fix & logging

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* add filter

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* fmt

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* u

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* convert may return tuple or dict :S

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

---------

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/diffusion/finetune.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/diffusion/finetune.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/diffusion/finetune.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/diffusion/finetune.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/diffusion/finetune.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/dataset-overview.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/overview.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Add cr changes

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* ci: Move source install fla to dev group (#1580)

* Move source install fla to dev group

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>

* Update uv lock

Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>

---------

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>
Co-authored-by: NeMo Bot <nemo-bot@nvidia.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update model coverage

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update overview doc

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update Hunyuan number of params

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Fix docs build issue

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

---------

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>
Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Co-authored-by: Andrew Chen <chenopis@users.noreply.github.com>
Co-authored-by: Piotr Żelasko <pzelasko@nvidia.com>
Co-authored-by: Dong Hyuk Chang <thomaschang26@tutanota.com>
Co-authored-by: NeMo Bot <nemo-bot@nvidia.com>
torsli pushed a commit that referenced this pull request Mar 24, 2026
* Move source install fla to dev group

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>

* Update uv lock

Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>

---------

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>
Co-authored-by: NeMo Bot <nemo-bot@nvidia.com>
torsli pushed a commit that referenced this pull request Mar 24, 2026
* feat: Integrate Wan with multi-resolution DL

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Required changes for compatability with AM container

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Fix overrides

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* feat: Add docs about diffusion support in AM

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Remove older data processing tools

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/diffusion/dataset.md

Co-authored-by: Andrew Chen <chenopis@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Apply suggestions from code review

Co-authored-by: Andrew Chen <chenopis@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Revert non-doc code changes to match main

This PR is docs-only; restore test and tool files to main's state.

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Restructure the diffusion finetuning doc

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* fix: Nemotron v3 inputs_embeds generation (#1583)

Fix Nemotron v3 inputs_embeds generation

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* fix: checkpointing for PEFT. (#1576)

* Fix checkpointing for PEFT.

Previously, the state_dict in the modelstate class had an if/elseif/else statement where
peft was handled in two caces and non-peft on the third one.

The first case of peft, was handling correctly, while the second was including
buffers causing issues in downstream consumers.

This fix simplifies the logic (simple if/else) and bypassed the issues with the
buffer.

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* Update stateful_wrappers.py

* fix

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* update test

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* improve error logging in test; pass is_peft to optimizerstate

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* fix

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* fix & logging

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* add filter

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* fmt

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* u

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* convert may return tuple or dict :S

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

---------

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/diffusion/finetune.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/diffusion/finetune.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/diffusion/finetune.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/diffusion/finetune.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/diffusion/finetune.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/dataset-overview.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/overview.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Add cr changes

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* ci: Move source install fla to dev group (#1580)

* Move source install fla to dev group

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>

* Update uv lock

Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>

---------

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>
Co-authored-by: NeMo Bot <nemo-bot@nvidia.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update model coverage

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update overview doc

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update Hunyuan number of params

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Fix docs build issue

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

---------

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>
Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Co-authored-by: Andrew Chen <chenopis@users.noreply.github.com>
Co-authored-by: Piotr Żelasko <pzelasko@nvidia.com>
Co-authored-by: Dong Hyuk Chang <thomaschang26@tutanota.com>
Co-authored-by: NeMo Bot <nemo-bot@nvidia.com>
zpqiu added a commit that referenced this pull request Apr 17, 2026
Move flash-linear-attention from the [dependency-groups].dev group
(PEP 735, resolvable only by this repo's own `uv sync`) into a dedicated
`fla` [project.optional-dependencies] extra, and compose it into the
`moe` and `all` extras. Downstream consumers (e.g. NeMo-RL) pull
nemo-automodel via the `pkg[extra]` syntax, which cannot reference
PEP 735 dependency-groups, so after #1580 `fla` was silently missing
for them.

Also drop the VCS SHA pin in favor of the official 0.4.2 release, now
that `fla.ops.cp` (required for CP on linear attention) is included
upstream.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
Signed-off-by: Zhaopeng Qiu <alexq@nvidia.com>
akoumpa added a commit that referenced this pull request Apr 20, 2026
* build: move flash-linear-attention back to optional-dependencies

Move flash-linear-attention from the [dependency-groups].dev group
(PEP 735, resolvable only by this repo's own `uv sync`) into a dedicated
`fla` [project.optional-dependencies] extra, and compose it into the
`moe` and `all` extras. Downstream consumers (e.g. NeMo-RL) pull
nemo-automodel via the `pkg[extra]` syntax, which cannot reference
PEP 735 dependency-groups, so after #1580 `fla` was silently missing
for them.

Also drop the VCS SHA pin in favor of the official 0.4.2 release, now
that `fla.ops.cp` (required for CP on linear attention) is included
upstream.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
Signed-off-by: Zhaopeng Qiu <alexq@nvidia.com>

* Update uv lock

Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>

---------

Signed-off-by: Zhaopeng Qiu <alexq@nvidia.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>
Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
Co-authored-by: NeMo Bot <nemo-bot@nvidia.com>
Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
linnanwang pushed a commit that referenced this pull request Apr 24, 2026
* Move source install fla to dev group

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>

* Update uv lock

Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>

---------

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>
Co-authored-by: NeMo Bot <nemo-bot@nvidia.com>
linnanwang pushed a commit that referenced this pull request Apr 24, 2026
* feat: Integrate Wan with multi-resolution DL

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Required changes for compatability with AM container

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Fix overrides

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* feat: Add docs about diffusion support in AM

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Remove older data processing tools

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/diffusion/dataset.md

Co-authored-by: Andrew Chen <chenopis@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Apply suggestions from code review

Co-authored-by: Andrew Chen <chenopis@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Revert non-doc code changes to match main

This PR is docs-only; restore test and tool files to main's state.

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Restructure the diffusion finetuning doc

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* fix: Nemotron v3 inputs_embeds generation (#1583)

Fix Nemotron v3 inputs_embeds generation

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* fix: checkpointing for PEFT. (#1576)

* Fix checkpointing for PEFT.

Previously, the state_dict in the modelstate class had an if/elseif/else statement where
peft was handled in two caces and non-peft on the third one.

The first case of peft, was handling correctly, while the second was including
buffers causing issues in downstream consumers.

This fix simplifies the logic (simple if/else) and bypassed the issues with the
buffer.

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* Update stateful_wrappers.py

* fix

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* update test

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* improve error logging in test; pass is_peft to optimizerstate

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* fix

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* fix & logging

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* add filter

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* fmt

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* u

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* convert may return tuple or dict :S

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

---------

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/diffusion/finetune.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/diffusion/finetune.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/diffusion/finetune.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/diffusion/finetune.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/diffusion/finetune.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/dataset-overview.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update docs/guides/overview.md

Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Add cr changes

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* ci: Move source install fla to dev group (#1580)

* Move source install fla to dev group

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>

* Update uv lock

Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>

---------

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>
Co-authored-by: NeMo Bot <nemo-bot@nvidia.com>
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update model coverage

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update overview doc

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Update Hunyuan number of params

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

* Fix docs build issue

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>

---------

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>
Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Co-authored-by: Andrew Chen <chenopis@users.noreply.github.com>
Co-authored-by: Piotr Żelasko <pzelasko@nvidia.com>
Co-authored-by: Dong Hyuk Chang <thomaschang26@tutanota.com>
Co-authored-by: NeMo Bot <nemo-bot@nvidia.com>
linnanwang pushed a commit that referenced this pull request Apr 24, 2026
* build: move flash-linear-attention back to optional-dependencies

Move flash-linear-attention from the [dependency-groups].dev group
(PEP 735, resolvable only by this repo's own `uv sync`) into a dedicated
`fla` [project.optional-dependencies] extra, and compose it into the
`moe` and `all` extras. Downstream consumers (e.g. NeMo-RL) pull
nemo-automodel via the `pkg[extra]` syntax, which cannot reference
PEP 735 dependency-groups, so after #1580 `fla` was silently missing
for them.

Also drop the VCS SHA pin in favor of the official 0.4.2 release, now
that `fla.ops.cp` (required for CP on linear attention) is included
upstream.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
Signed-off-by: Zhaopeng Qiu <alexq@nvidia.com>

* Update uv lock

Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>

---------

Signed-off-by: Zhaopeng Qiu <alexq@nvidia.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>
Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
Co-authored-by: NeMo Bot <nemo-bot@nvidia.com>
Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants