transformers/tests/models/wav2vec2
Matt edb704b26e
Fix inverted conditional in TF common test! (#22540)
* Fix inverted conditional in TF common test!

* Make the same change in the PT tests file

* Make sure hidden states for GPT2 have the same output shape in PT/TF

* Minor fix to PT implementation of token classification loss

* Skip loss equivalence test for TFHubert because it keeps overflowing to inf

* Compute LM loss for TF the (weird) way it's computed in PT

* Skip loss equivalence test for Wav2Vec2 for the same reason as Hubert

* Fix - don't try to access the hidden states property when output is a tuple
2023-04-04 21:59:54 +01:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_feature_extraction_wav2vec2.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_modeling_flax_wav2vec2.py Fix env. variable type issue in testing (#21609) 2023-02-13 20:53:26 +01:00
test_modeling_tf_wav2vec2.py Fix inverted conditional in TF common test! (#22540) 2023-04-04 21:59:54 +01:00
test_modeling_wav2vec2.py Time to Say Goodbye, torch 1.7 and 1.8 (#22291) 2023-03-21 19:22:01 +01:00
test_processor_wav2vec2.py [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00:00
test_tokenization_wav2vec2.py Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00