PR #27220 Fix switch transformer mixed precision issue

Fix switch transformer mixed precision issue #27220

asartran merged 2 commits into huggingface:main from fix-switch-transformer-mixed-precision

Fix mixed precision error for switch transformer

cd121a59

Fixup

5cad1b0f

younesbelkada approved these changes on 2023-11-02

younesbelkada requested a review from

asartran 3 years ago

asartran approved these changes on 2023-11-02

younesbelkada commented on 2023-11-02

asartran merged f13f544a into main 3 years ago

Reviewers

younesbelkada

asartran

Assignees

No one assigned

Labels

None yet

Milestone

No milestone