Kian Sierra McGettigan 6bf885375a Prophetnet batch dimension inversion fix (#21870) ...

* decoder forward pass is working

* no model has forward pass returning attentions

* decoder ngram changed to not mix batch size

* current basic forward pass returns identical result

* passed test_model attentions

* passed test_encoder_decoder_model_generate

* passed test_headmasking

* removed old block

* removed comments bug/fixme

* removed bug comments

* applied styling

* applied fix-copies

* applied ngram forward comments

* corrected dimension notation

* applied styling and comment fixes

* changed asserts for raise ValueError

* changed question gen test

* updated hidden_states integration test

* applied styling

2023-03-02 12:07:45 -05:00

..

__init__.py

Move test model folders (#17034)

2022-05-03 14:42:02 +02:00

test_modeling_prophetnet.py

Prophetnet batch dimension inversion fix (#21870)

2023-03-02 12:07:45 -05:00

test_tokenization_prophetnet.py

Update quality tooling for formatting (#21480)

2023-02-06 18:10:56 -05:00