Convert Flax Bart 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 Bart to a standalone model script #2

Closed sanchit-gandhi closed 2 years ago

sanchit-gandhi commented 2 years ago

Convert modeling_flax_bart from a model file situated within Transformers to a functional standalone model script. This PR also dispenses of all irrelevant Flax Bart modules for the focus of this project (e.g. EncoderModule, ForConditionalGeneration, ForQuestionAnswering).