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

centralize place for defining block #68

Merged
merged 8 commits into from
Apr 2, 2024
Merged

centralize place for defining block #68

merged 8 commits into from
Apr 2, 2024

Conversation

lchu6
Copy link
Contributor

@lchu6 lchu6 commented Apr 2, 2024

Currently transformer block is defined and used in a few places including ac_handler module and fsdp_wrapper module.
This PR will centralize these into main (where the model is defined) so it is easier to switch from one model to another.

@lchu6 lchu6 self-assigned this Apr 2, 2024
@lchu6 lchu6 marked this pull request as draft April 2, 2024 19:07
@lchu6 lchu6 marked this pull request as ready for review April 2, 2024 19:19
@lchu6 lchu6 merged commit 5482493 into main Apr 2, 2024
3 checks passed
@lchu6 lchu6 deleted the unify branch April 2, 2024 20:28
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.

2 participants