Skip to content

feat: add embeddings utils funcs to not added models - #1288

Merged
thomasdhc merged 2 commits into
NVIDIA-NeMo:mainfrom
SwekeR-463:feat/add-embed-utils
Feb 17, 2026
Merged

thomasdhc merged 2 commits into
NVIDIA-NeMo:mainfrom
SwekeR-463:feat/add-embed-utils

Conversation

@SwekeR-463

@SwekeR-463 SwekeR-463 commented Feb 14, 2026 •

Copy link
Copy Markdown
Contributor

What does this PR do ?

  • Added get_input_embeddings(), set_input_embeddings(), get_output_embeddings(), and set_output_embeddings() methods to models that were missing them.

Changelog

  • deepseek_v3, deepseek_v32, glm4_moe, glm4_moe_lite, gpt_oss, minimax_m2, mistral3, nemotron_v3, qwen2, qwen3_moe, qwen3_next, step3p5: Added all 4 methods
  • qwen3_omni_moe: Added get_output_embeddings, set_output_embeddings
  • qwen3_vl_moe: Added get_output_embeddings, set_output_embeddings
  • nemotron_parse: Added set_input_embeddings

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?

Additional Information

@copy-pr-bot

copy-pr-bot Bot commented Feb 14, 2026

Copy link
Copy Markdown

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.

Signed-off-by: SwekeR-463 <swekerswasti@gmail.com>
Signed-off-by: SwekeR-463 <swekerswasti@gmail.com>
@akoumpa

akoumpa commented Feb 14, 2026

Copy link
Copy Markdown
Contributor

/ok to test 5619efe

@akoumpa akoumpa 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.

LGTM, thanks a lot @SwekeR-463

@akoumpa

akoumpa commented Feb 14, 2026

Copy link
Copy Markdown
Contributor

@NVIDIA-NeMo/automation this is ok to FM, please take a look when you get the chance. Thank you.

@chtruong814 chtruong814 added the needs-follow-up Issue needs follow-up label Feb 16, 2026

@thomasdhc thomasdhc 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.

Bypass codecov

@thomasdhc
thomasdhc merged commit 7d35447 into NVIDIA-NeMo:main Feb 17, 2026
52 of 53 checks passed
@chtruong814 chtruong814 removed the needs-follow-up Issue needs follow-up label Feb 17, 2026
akoumpa pushed a commit that referenced this pull request Feb 25, 2026
* feat: add i/o embeddings get and set funcs to not added models

Signed-off-by: SwekeR-463 <swekerswasti@gmail.com>

* ..

Signed-off-by: SwekeR-463 <swekerswasti@gmail.com>

---------

Signed-off-by: SwekeR-463 <swekerswasti@gmail.com>
linnanwang pushed a commit that referenced this pull request Apr 24, 2026
* feat: add i/o embeddings get and set funcs to not added models

Signed-off-by: SwekeR-463 <swekerswasti@gmail.com>

* ..

Signed-off-by: SwekeR-463 <swekerswasti@gmail.com>

---------

Signed-off-by: SwekeR-463 <swekerswasti@gmail.com>
kashif pushed a commit to kashif/Automodel that referenced this pull request May 25, 2026
* feat: add i/o embeddings get and set funcs to not added models

Signed-off-by: SwekeR-463 <swekerswasti@gmail.com>

* ..

Signed-off-by: SwekeR-463 <swekerswasti@gmail.com>

---------

Signed-off-by: SwekeR-463 <swekerswasti@gmail.com>

This branch was previously deployed

2 inactive deployments
nemo-ci — 5619efeb Deployed Feb 14, 2026 by copy-pr-bot[bot] via L2_HF_Transformer_Finetune #3845
test — 5619efeb Deployed Feb 14, 2026 by copy-pr-bot[bot] via cicd-wait-in-queue #3845
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants