-
Notifications
You must be signed in to change notification settings - Fork 304
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
[Multimodal model] Add the decoder of the multimodal model #626
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
As discussed offline, let's create separate files (in the folder) for encoder, decoder, and shared parts. This would make code more readable than a 1.5k LoC monolithic file.
I had a similar suggestion for ModelArgs
, since it seems to be shared by encoder and decoder.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good to me. Please update the ModelArgs
before merge.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Nice work!
After pytorch#589, we are adding decoder model to torchtitan with simple unit test.
After #589, we are adding decoder model to torchtitan with simple unit test.