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.

Sign up for GitHub

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

Jump to bottom

ESM2 Finetuning refactor #574

Open
wants to merge 12 commits into
base: main
Choose a base branch
from
Open

ESM2 Finetuning refactor #574

wants to merge 12 commits into from
+1,757 −776

Conversation

farhadrgh
Copy link
Collaborator

@farhadrgh farhadrgh commented Jan 6, 2025 •
edited
Loading

Description

Type of changes

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Refactor
  • Documentation update
  • Other (please describe):

CI Pipeline Configuration

Configure CI behavior by checking relevant boxes below. This will automatically apply labels.

Note

By default, the notebooks validation tests are skipped unless explicitly enabled.

Usage

TODO: Add code snippet

Pre-submit Checklist

  • I have tested these changes locally
  • I have updated the documentation accordingly
  • I have added/updated tests as needed
  • All existing tests pass successfully

@farhadrgh
refactor datasets
22bb887 
Signed-off-by: Farhad Ramezanghorbani <farhadr@nvidia.com>
@farhadrgh
refactor datasets
4848f9c 
Signed-off-by: Farhad Ramezanghorbani <farhadr@nvidia.com>
@farhadrgh
add finetune script
db955ef 
Signed-off-by: Farhad Ramezanghorbani <farhadr@nvidia.com>
@farhadrgh
Copy link
Collaborator Author

Ported over the changes from #546

farhadrgh and others added 9 commits January 6, 2025 12:52
@farhadrgh
fix typing
4cac6d7 
Signed-off-by: Farhad Ramezanghorbani <farhadr@nvidia.com>
@farhadrgh
simplify, use 650m
900ef00 
Signed-off-by: Farhad Ramezanghorbani <farhadr@nvidia.com>
@farhadrgh
Merge branch 'main' into farhadr/ft_refactor
d7d526b
@farhadrgh
executable finetune_esm2
5e2dd4b 
Signed-off-by: Farhad Ramezanghorbani <farhadr@nvidia.com>
@farhadrgh
typing
4577bc2 
Signed-off-by: Farhad Ramezanghorbani <farhadr@nvidia.com>
@farhadrgh
resolve conflicts
9154c56
@farhadrgh
add finetune notebook
8fc64a3 
Signed-off-by: Farhad Ramezanghorbani <farhadr@nvidia.com>
@farhadrgh
move test
087341a 
Signed-off-by: Farhad Ramezanghorbani <farhadr@nvidia.com>
@farhadrgh
deprecate old example
3eb8740 
Signed-off-by: Farhad Ramezanghorbani <farhadr@nvidia.com>
@farhadrgh farhadrgh marked this pull request as ready for review January 9, 2025 17:58
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@jstjohn jstjohn Awaiting requested review from jstjohn jstjohn is a code owner

@malcolmgreaves malcolmgreaves Awaiting requested review from malcolmgreaves malcolmgreaves is a code owner

@skothenhill-nv skothenhill-nv Awaiting requested review from skothenhill-nv skothenhill-nv is a code owner

@ohadmo ohadmo Awaiting requested review from ohadmo ohadmo is a code owner

@pstjohn pstjohn Awaiting requested review from pstjohn pstjohn is a code owner

@trvachov trvachov Awaiting requested review from trvachov trvachov is a code owner

@dorotat-nv dorotat-nv Awaiting requested review from dorotat-nv dorotat-nv is a code owner

At least 2 approving reviews are required to merge this pull request.

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@farhadrgh