transformers/tests/models/wav2vec2
Ritik Nandwal 20ac86c6f1
Add TensorFlow Wav2Vec2 for sequence classification (#22073)
* Add initial changes for TF wav2vec2 for sequence classification

* Add suggested changes

* Add serving and serving output methods

* Add serving_output implementation and fix layer_weights

* Add fixes

* Fixed test cases

* Fixing test and adding suggested changes
2023-04-26 13:35:30 +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 Add TensorFlow Wav2Vec2 for sequence classification (#22073) 2023-04-26 13:35:30 +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