• Patrick Fernandes's avatar
    [Flax] Add Electra models (#11426) · 0afe4a90
    Patrick Fernandes authored
    
    
    * add electra model to flax
    
    * Remove Electra Next Sentence Prediction model added by mistake
    
    * fix parameter sharing and loosen equality threshold
    
    * fix styling issues
    
    * add mistaken removen imports
    
    * fix electra table
    
    * Add FlaxElectra to automodels and fixe docs
    
    * fix issues pointed out the PR
    
    * fix flax electra to comply with latest changes
    
    * remove stale class
    
    * add copied from
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    0afe4a90
index.rst 36.7 KB