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

Add support for Transformer models #45

Open
zhe-meng opened this issue Jun 9, 2021 · 2 comments
Open

Add support for Transformer models #45

zhe-meng opened this issue Jun 9, 2021 · 2 comments
Assignees
Labels
help wanted Extra attention is needed module: modules Related to modules type: improvement New feature or request
Milestone

Comments

@zhe-meng
Copy link

zhe-meng commented Jun 9, 2021

Hi, thanks for your work. Is it possible to add support for the transformer models, especifically layers implemented with the einops library.

@zhe-meng zhe-meng added the type: improvement New feature or request label Jun 9, 2021
@frgfm frgfm changed the title Hi, thanks for your work. Is it possible to add support for the transformer models, especifically layers implemented with the einops library. Add support for Transformer models Jun 9, 2021
@frgfm frgfm added this to the 0.1.2 milestone Jun 9, 2021
@frgfm
Copy link
Owner

frgfm commented Jun 13, 2021

Hi @zhe-meng,

Thanks for your very relevant feature request!
I agree that support should be added to the library, I'll try to work on this soon. If you have some ideas about a PR, feel free to open one in the meantime!

@frgfm
Copy link
Owner

frgfm commented Jul 11, 2021

Hey @zhe-meng,

Small update, I started working on this and it will be handled in a few PRs:

I hope it will help 👍

@frgfm frgfm added the help wanted Extra attention is needed label May 18, 2022
@frgfm frgfm modified the milestones: 0.1.2, 0.2.0 Aug 3, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
help wanted Extra attention is needed module: modules Related to modules type: improvement New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants