Skip to content
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

[DRAFT] Support custom recipes in NeMoRun #325

Draft
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

TaekyungHeo
Copy link
Member

@TaekyungHeo TaekyungHeo commented Jan 8, 2025

Summary

This PR introduces support for custom recipes in NeMoRun, as requested by Itamar. NeMoRun allows users to register recipes at https://github.com/NVIDIA/NeMo/tree/main/nemo/collections/llm/recipes. Once a recipe is registered, it becomes available via the NeMoRun command (nemo). To register a custom recipe, the following steps should be done:

  • Add the new recipe to /opt/NeMo/nemo/collections/llm/recipes/.
  • Update /opt/NeMo/nemo/collections/llm/recipes/init.py (add import and update __all__).

TODO: How to update /opt/NeMo/nemo/collections/llm/recipes/init.py?

Test Plan

  • Added unit tests.
  • Verified functionality with a custom recipe.

@TaekyungHeo TaekyungHeo changed the title Support custom recipes in NeMoRun [DRAFT] Support custom recipes in NeMoRun Jan 8, 2025
@TaekyungHeo TaekyungHeo added the enhancement New feature or request label Jan 8, 2025
@TaekyungHeo TaekyungHeo force-pushed the nemo-run-custom-recipe branch from fad7241 to 1f702f0 Compare January 8, 2025 11:23
@TaekyungHeo TaekyungHeo force-pushed the nemo-run-custom-recipe branch from 1f702f0 to 7f17e2f Compare January 8, 2025 11:29
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant