transformers
9a3f4d4d
- Bark model Flash Attention 2 Enabling to pass on check_device_map parameter to super() (#29357)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Bark model Flash Attention 2 Enabling to pass on check_device_map parameter to super() (#29357) * Fixing error #29332. The _check_and_enable_flash_attn_2() method receives a check_device_map parameter and fails. * style fixup
References
#29357 - Bark model Flash Attention 2 Enabling to pass on check_device_map parameter to super()
Author
damithsenanayake
Parents
6d67837f
Loading