Sanchit Gandhi
a960406722
[FlaxWav2Vec2Model] Fix bug in attention mask ( #16725 )
...
* [FlaxWav2Vec2Model] Fix bug in attention mask
* more fixes
* add (Flax)SpeechEncoderDecoderModel PT-FX cross-test
2022-04-12 19:48:24 +02:00
Sanchit Gandhi
8d57c424e0
[FlaxSpeechEncoderDecoderModel] More Rigorous PT-Flax Equivalence Tests ( #16589 )
2022-04-06 15:33:32 +02:00
Sanchit Gandhi
7ca4633555
[FlaxSpeechEncoderDecoderModel] Ensure Input and Output Word Embeddings Are **Not** Tied ( #16444 )
...
* [FlaxSpeechEncoderDecoderModel] Ensure Input and Output Word Embeddings Are **Not** Tied
* rebase
2022-03-28 14:14:10 +02:00
Sanchit Gandhi
925fc57b70
[Flax] Improve Robustness of Back-Prop Tests ( #16418 )
...
* [Flax] Improve Robustness of Back-Prop Tests
* check equality of logits/outputs
* make fixup
2022-03-28 11:56:54 +02:00
Sanchit Gandhi
e231c72906
[FlaxSpeechEncoderDecoder] Fix feature extractor gradient test ( #16407 )
2022-03-25 17:46:53 +01:00
Suraj Patil
b25b92ac4f
update jax version and re-enable some tests ( #16254 )
2022-03-18 16:45:39 +01:00
Suraj Patil
632ff3c39e
[FlaxSpeechEncoderDecoderModel] Skip from_encoder_decoder_pretrained ( #16236 )
...
* skip the test
* fix
* fix skip
2022-03-17 20:05:14 +01:00
Sanchit Gandhi
2de99e6c43
Fix Loading of Flax(Speech)EncoderDecoderModel kwargs from PreTrained Encoder-Decoder Checkpoints ( #16056 )
...
* Fix Loading of Flax(Speech)EncoderDecoderModel kwargs from PreTrained Encoder-Decoder Checkpoints
* change wording
2022-03-14 10:12:29 +01:00
Sanchit Gandhi
1da84ae02c
Fix Bug in Flax-Speech-Encoder-Decoder Test ( #16041 )
...
* Fix Bug in Flax-Speech-Encoder-Decoder Test
* change thresholds for CPU precision
2022-03-10 12:09:29 +01:00
Sanchit Gandhi
fde901877a
Freeze Feature Encoder in FlaxSpeechEncoderDecoder ( #15997 )
...
* Freeze Feature Encoder in FlaxSpeechEncoderDecoder
* add backprop test
2022-03-10 09:59:19 +01:00
Sanchit Gandhi
b256f3518d
Add FlaxBartForCausalLM ( #15995 )
...
* add causal lm
* add CausalLM tests
* Add FlaxBartForCausalLM
* Add EncoderDecoder model tests
* change docstring
* make repo-consistency
* suggested changes
* remove jax ops
* correction
* rename pre-trained decoder model
2022-03-09 19:53:01 +01:00
Sanchit Gandhi
e3342edc4e
Flax Speech-Encoder-Decoder Model ( #15613 )
...
* rebase
* Delete shift tokens func
* downsample decoder input seq len for init
* correct attention mask
* add tests
* pt flax cross test
* make fixup
* init file for import
* change pt-flax cross test threshold
* pt-flax test logits only
* move tests
* make repo-consistency
* consistent indentation
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2022-02-28 12:22:36 +01:00
Lysandre Debut
29c10a41d0
[Test refactor 1/5] Per-folder tests reorganization ( #15725 )
...
* Per-folder tests reorganization
Co-authored-by: sgugger <sylvain.gugger@gmail.com>
Co-authored-by: Stas Bekman <stas@stason.org>
2022-02-23 15:46:28 -05:00
