transformers
36778030
- Fix accelerate hooks for all models using linear attention (#46978)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 days ago
Fix accelerate hooks for all models using linear attention (#46978) * add wrapper * fix modular * move to generics * add all mambas * much better with decorator * oupsi * move to accelerate * fix imports * add test
References
#46978 - Fix accelerate hooks for all models using linear attention
Author
Cyrilvallez
Parents
17518d53
Loading