transformers
4dd59f71 - [Nit] Add kernels_fallback_ok kwarg to is_flash_attn_N_available (#47318)

Commit

1 day ago

[Nit] Add kernels_fallback_ok kwarg to is_flash_attn_N_available (#47318) * Core idea * Smol fixes * Style * Name * Restore guard * Fix * Name nits * Added tests

References

#47318 - [Nit] Add kernels_fallback_ok kwarg to is_flash_attn_N_available

Author

remi-or

Parents

b7f01015

transformers 4dd59f71 - [Nit] Add kernels_fallback_ok kwarg to is_flash_attn_N_available (#47318)

transformers
4dd59f71 - [Nit] Add kernels_fallback_ok kwarg to is_flash_attn_N_available (#47318)