• Crystina's avatar
    Flax t5 Encoder (#17784) · 692e61e9
    Crystina authored
    
    
    * first draft adding Flax-t5-encoder and Flax-mt5-encoder
    
    * imports
    
    * after make fixup
    
    * flax t5 encoder test
    
    * black on test
    
    * make fix-copies
    
    * clean
    
    * all_model_classes -> tuple
    
    * clean test
    
    * is_encoder_decoder=False in t5-enc tester
    
    * remove file docstring before FlaxT5Encoder
    
    * black
    
    * isort
    
    * commit suggestions on src/transformers/models/t5/modeling_flax_t5.py
    Co-authored-by: default avatarSuraj Patil <surajp815@gmail.com>
    
    * commit suggestions on src/transformers/models/t5/modeling_flax_t5.py
    Co-authored-by: default avatarSuraj Patil <surajp815@gmail.com>
    
    * Apply suggestions from code review
    Co-authored-by: default avatarSuraj Patil <surajp815@gmail.com>
    
    * remove _get_encoder_module
    
    * self.decoder_seq_length -> self.encoder_seq_length as t5-enc does not have decoder
    
    * bugfix - self.module_class is class itself, not instance;
    
    * docs for mt5 and t5
    
    * call -> __call__ in t5 doc
    
    * FlaxMT5EncoderModel to TYPE_HINT
    
    * run doc-builder to allow change the files
    Co-authored-by: default avatarSuraj Patil <surajp815@gmail.com>
    692e61e9
test_modeling_flax_t5.py 57.5 KB