tests/models/wav2vec2/test_modeling_flax_wav2vec2.py · af150e4a1c4c4efef634bc5ee3a1773a97772ffb · 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_flax_wav2vec2.py 24.7 KB

Replace test_modeling_flax_wav2vec2.py