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

#3974: nanogpt uplift and move weights to weka path #4221

Merged
merged 1 commit into from
Jan 19, 2024
Merged

#3974: nanogpt uplift and move weights to weka path #4221

merged 1 commit into from
Jan 19, 2024

Conversation

punithsekar
Copy link
Contributor

@punithsekar punithsekar commented Dec 7, 2023 •
edited
Loading

This PR has following

  • Uplifted nanoGPT model and recorded the ops that do not have TT-ops
  • Tilized weights tesnors are stored and loaded from weka path.
  • Tests for both sub-modules and the whole model.

@punithsekar punithsekar force-pushed the punith/uplift_nanogpt branch 2 times, most recently from c4163c7 to aca0230 Compare December 7, 2023 12:12
@punithsekar punithsekar mentioned this pull request Dec 7, 2023
Open
@saichandax
Copy link
Contributor

CI passed. Link here

@saichandax saichandax merged commit 9ae0ed0 into main Jan 19, 2024
4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@farbabi farbabi farbabi approved these changes

@boris-drazic boris-drazic boris-drazic approved these changes

@Sudharsan-V Sudharsan-V Awaiting requested review from Sudharsan-V

@saichandax saichandax Awaiting requested review from saichandax

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

Successfully merging this pull request may close these issues.
4 participants
@punithsekar @saichandax @boris-drazic @farbabi