transformers
Bark model Flash Attention 2 Enabling to pass on check_device_map parameter to super()
#29357

Merged

Bark model Flash Attention 2 Enabling to pass on check_device_map parameter to super() #29357

amyeroberts merged 2 commits into huggingface:main from damithsenanayake:flash_attn_dev_mp_check

ArthurZucker commented on 2024-03-07

damithsenanayake force pushed 2 years ago

Fixing error #29332. The _check_and_enable_flash_attn_2() method rece…

95469048

style fixup

09ddd773

damithsenanayake force pushed to 09ddd773 2 years ago

amyeroberts approved these changes on 2024-03-11

amyeroberts merged 9a3f4d4d into main 2 years ago

Reviewers

amyeroberts

ArthurZucker

Assignees

No one assigned

Labels

None yet

Milestone

No milestone