8000 Added safe loading of models by nithinraok · Pull Request #13607 · NVIDIA/NeMo · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

Added safe loading of models #13607

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.

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

Merged
merged 6 commits into from
Jun 9, 2025
Merged

Added safe loading of models #13607

merged 6 commits into from
Jun 9, 2025

Conversation

nithinraok
Copy link
Collaborator
@nithinraok nithinraok commented May 15, 2025

Important

The Update branch button must only be pressed in very rare occassions.
An outdated branch is never blocking the merge of a PR.
Please reach out to the automation team before pressing that button.

What does this PR do ?

This PR addresses known security issues. For the latest NVIDIA Vulnerability Disclosure Information visit https://www.nvidia.com/en-us/security/,

Usage

  • You can potentially add a usage example below
# Add a code snippet demonstrating how to use this 

GitHub Actions CI

The Jenkins CI system has been replaced by GitHub Actions self-hosted runners.

The GitHub Actions CI will run automatically when the "Run CICD" label is added to the PR.
To re-run CI remove and add the label again.
To run CI on an untrusted fork, a NeMo user with write access must first click "Approve and run".

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?
  • Does the PR affect components that are optional to install? (Ex: Numba, Pynini, Apex etc)
    • Reviewer: Does the PR have correct import guards for all optional libraries?

PR Type:

  • New Feature
  • Bugfix
  • Documentation

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

Who can review?

Anyone in the community is free to review the PR once the checks have passed.
Contributor guidelines contains specific people who can review PRs to various areas.

Additional Information

  • Related to # (issue)

@github-actions github-actions bot added the core Changes to NeMo Core label May 15, 2025
@nithinraok nithinraok requested a review from chtruong814 May 15, 2025 19:33
chtruong814
chtruong814 previously approved these changes May 15, 2025
@chtruong814
Copy link
Collaborator

We're going to skip the Eval unit test that is failing for now. The flux failure is likely flaky. But the speechLM failure seems like we need to add another module path to the allow list. At least for the test.

Nithin Rao Koluguri and others added 4 commits May 19, 2025 13:48
Signed-off-by: Nithin Rao Koluguri <nithinraok>
Signed-off-by: Nithin Rao Koluguri <nithinraok>
Signed-off-by: nithinraok <nithinraok@users.noreply.github.com>
Signed-off-by: Nithin Rao Koluguri <nithinraok>
Signed-off-by: Nithin Rao Koluguri <nithinraok>
@github-actions github-actions bot removed the Run CICD label May 29, 2025
Signed-off-by: nithinraok <nithinrao.koluguri@gmail.com>
Copy link
Contributor
github-actions bot commented Jun 5, 2025

[🤖]: Hi @nithinraok 👋,

We wanted to let you know that a CICD pipeline for this PR just finished successfully.

So it might be time to merge this PR or get some approvals.

//cc @chtruong814 @ko3n1g @pablo-garay @thomasdhc

@nithinraok
Copy link
Collaborator Author

@ericharper could you review this PR please

Copy link
Collaborator
@ericharper ericharper left a comment

Choose a reason for hiding this comment

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

LGTM. Thanks!

@ericharper ericharper merged commit 121456e into main Jun 9, 2025
282 of 290 checks passed
@ericharper ericharper deleted the address_hydra_instantiation branch June 9, 2025 20:40
gdengk pushed a commit to gdengk/NeMo that referenced this pull request Jun 12, 2025
* Added safe loading of models

Signed-off-by: Nithin Rao Koluguri <nithinraok>

* add Any typing

Signed-off-by: Nithin Rao Koluguri <nithinraok>

* Apply isort and black reformatting

Signed-off-by: nithinraok <nithinraok@users.noreply.github.com>

* add tests folder

Signed-off-by: Nithin Rao Koluguri <nithinraok>

* add lightning module

Signed-off-by: Nithin Rao Koluguri <nithinraok>

* add megatron

Signed-off-by: nithinraok <nithinrao.koluguri@gmail.com>

---------

Signed-off-by: Nithin Rao Koluguri <nithinraok>
Signed-off-by: nithinraok <nithinraok@users.noreply.github.com>
Signed-off-by: nithinraok <nithinrao.koluguri@gmail.com>
Co-authored-by: Nithin Rao Koluguri <nithinraok>
Co-authored-by: nithinraok <nithinraok@users.noreply.github.com>
@chtruong814 chtruong814 added r2.4.0 Pick this label for auto-cherry-picking into r2.4.0 r2.3.0 Pick this label for auto-cherrypicking into v2.3.0 and removed r2.4.0 Pick this label for auto-cherry-picking into r2.4.0 labels Jun 22, 2025
chtruong814 pushed a commit that referenced this pull request Jun 22, 2025
* Added safe loading of models

Signed-off-by: Nithin Rao Koluguri <nithinraok>

* add Any typing

Signed-off-by: Nithin Rao Koluguri <nithinraok>

* Apply isort and black reformatting

Signed-off-by: nithinraok <nithinraok@users.noreply.github.com>

* add tests folder

Signed-off-by: Nithin Rao Koluguri <nithinraok>

* add lightning module

Signed-off-by: Nithin Rao Koluguri <nithinraok>

* add megatron

Signed-off-by: nithinraok <nithinrao.koluguri@gmail.com>

---------

Signed-off-by: Nithin Rao Koluguri <nithinraok>
Signed-off-by: nithinraok <nithinraok@users.noreply.github.com>
Signed-off-by: nithinraok <nithinrao.koluguri@gmail.com>
Co-authored-by: Nithin Rao Koluguri <nithinraok>
Co-authored-by: nithinraok <nithinraok@users.noreply.github.com>
Signed-off-by: Charlie Truong <chtruong@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
core Changes to NeMo Core r2.3.0 Pick this label for auto-cherrypicking into v2.3.0 skip-docs skip-linting
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants
0