transformers

5f0d07b3 - Make BigBird model compatiable to fp16 dtype. (#16034)

Make BigBird model compatiable to fp16 dtype. (#16034) * Make BigBird model compatiable to fp16 dtype. * Use tree_map instead of map * Reformat the code * Fix import order * Convert masks to the correct dtype * Fix format issue * Address comments.