Bhadresh Savani c1e47bf4fe [Flax] Addition of FlaxPegasus (#13420) ...

* added initial files

* fixes pipeline

* fixes style and quality

* fixes doc issue and positional encoding

* fixes layer norm and test

* fixes quality issue

* fixes code quality

* removed extra layer norm

* added layer norm back in encoder and decoder

* added more code copy quality checks

* update tests

* Apply suggestions from code review

* fix import

* fix test

Co-authored-by: patil-suraj <surajp815@gmail.com>

2021-09-14 17:15:19 +02:00

15 KiB

Raw Blame History

View Raw