PR #16034 Make BigBird model compatiable to fp16 dtype.

Make BigBird model compatiable to fp16 dtype. #16034

patrickvonplaten merged 7 commits into huggingface:main from xuzhao9:xz9/fix-bigbird-fp16

patil-suraj requested a review from

patrickvonplaten 3 years ago

patrickvonplaten commented on 2022-03-14

xuzhao9 requested a review from

patrickvonplaten 3 years ago

patrickvonplaten commented on 2022-03-22

xuzhao9 force pushed 3 years ago

xuzhao9 requested a review from

patrickvonplaten 3 years ago

Make BigBird model compatiable to fp16 dtype.

d7a24ebf

Use tree_map instead of map

894207d1

Reformat the code

a7a1fef1

Fix import order

a8a3bb7c

Convert masks to the correct dtype

5e8b59b9

Fix format issue

caffd653

Address comments.

ca8e1bfb

xuzhao9 force pushed to ca8e1bfb 3 years ago

patrickvonplaten approved these changes on 2022-03-23

patrickvonplaten merged 5f0d07b3 into main 3 years ago

xuzhao9 deleted the xz9/fix-bigbird-fp16 branch 3 years ago

Reviewers

patrickvonplaten

Assignees

No one assigned

Labels

None yet

Milestone

No milestone