Patrick von Platen cee2d2135f [Flax Generation] Correct inconsistencies PyTorch/Flax (#12662) ...

* fix_torch_device_generate_test

* remove @

* correct greedy search

* save intertmed

* add final logits bias

* correct

* up

* add more tests

* fix another bug

* finish tests

* finish marian tests

* up

Co-authored-by: Patrick von Platen <patrick@huggingface.co>

2021-07-13 18:53:30 +01:00

18 KiB

Raw Blame History

View Raw