Daniel Stancl 21aecc0971 Add Flax RoFormer (#15005) ...

* Add FlaxRoFormer

* Clean code + make quality

* Fix output pooling for FlaxRoFormerForMultipleChoiceModule

* Apply suggestions from code review

* add flax model to repos

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

2022-01-04 13:23:10 +01:00

42 KiB

Raw Blame History

View Raw