Add memory efficient attention

SforAiDl / vformer

A modular PyTorch library for vision transformer models

https://vformer.readthedocs.io/

MIT License

162 stars 22 forks source link

Closed alvanli closed 2 years ago

alvanli commented 2 years ago

oop that's a lot of fails, I will fix them

abhi-glitchhg commented 2 years ago

Fixes #74

NeelayS commented 2 years ago

Looks pretty good @alvanli. Could you please also -

NeelayS commented 2 years ago

Merging. Thanks @alvanli!