tests/models/wav2vec2/test_modeling_tf_wav2vec2.py · f1e42bc50e47c66360fb9d2edf55c742ce6a8245 · chenpangpang / transformers

Allow user-managed Pool in Wav2Vec2ProcessorWithLM.batch_decode (#18351) · af150e4a

Antonio Carlos Falcão Petri authored Oct 18, 2022



* [Wav2Vec2] Allow user-managed Pool in Wav2Vec2ProcessorWithLM.batch_decode

* [Wav2Vec2] Add user-managed LM's pool tests and usage examples

* Improve styling
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

* [Wav2Vec2] Fix hyperlink references
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

af150e4a

test_modeling_tf_wav2vec2.py 26 KB

Replace test_modeling_tf_wav2vec2.py