facebookresearch / ToMe

A method to increase the speed and lower the memory footprint of existing vision transformers.
Other
931 stars 67 forks source link

Initial pass at ToMe-ing Stable Diffusion unets #6

Closed blindcrone closed 1 year ago

blindcrone commented 1 year ago

It has a roughly ~25% speedup in my experiments. This is less than I would have thought. What am I missing?

lalalune commented 1 year ago

Deprecated in favor of https://github.com/facebookresearch/ToMe/pull/7