Add support for Transformer models #45
Labels
help wanted
Extra attention is needed
module: modules
Related to modules
type: improvement
New feature or request
Milestone
Hi, thanks for your work. Is it possible to add support for the transformer models, especifically layers implemented with the einops library.
The text was updated successfully, but these errors were encountered: