Convert Flax Wav2Vec2 to a standalone model script

sanchit-gandhi / seq2seq-speech

Repository for fine-tuning Transformers 🤗 based seq2seq speech models in JAX/Flax.

34 stars 6 forks source link

Convert Flax Wav2Vec2 to a standalone model script #3

Closed sanchit-gandhi closed 2 years ago

sanchit-gandhi commented 2 years ago

Convert modeling_flax_wav2vec2 from a model file situated within Transformers to a functional standalone model script. This PR also dispenses of all irrelevant Flax Wav2Vec2 modules for the focus of this project (e.g. ForCTC, ForPreTraining).