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

Norms registry #1080

Merged
merged 16 commits into from
Apr 5, 2024
Merged

Norms registry #1080

merged 16 commits into from
Apr 5, 2024

Conversation

dakinggg
Copy link
Collaborator

@dakinggg dakinggg commented Apr 2, 2024

Starting the layers part of the registry with just the norms so we can review the overall setup of the layers registry in a smaller PR.

Note: In order to get the type checker to be happy with the decorator version of registering classes, I had to add a register_class method with more scoped typing. Open to other suggestions for how to accomplish this.

Screenshot 2024-04-02 at 12 00 50 AM

Also manually tested that the hf checkpoint saved can be loaded back in with AutoModelForCausalLM

@dakinggg dakinggg marked this pull request as ready for review April 2, 2024 08:06
@dakinggg dakinggg requested a review from a team as a code owner April 2, 2024 08:06
Copy link
Contributor

@irenedea irenedea left a comment

Choose a reason for hiding this comment

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

lgtm!

@dakinggg dakinggg merged commit b81897a into mosaicml:main Apr 5, 2024
9 checks passed
KuuCi pushed a commit that referenced this pull request Apr 18, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants