Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Julien Plu's avatar
    Restore TF embeddings and attention layers to their previous version (#9890) · 31563e05
    Julien Plu authored 
    * Refacto BERT

* Restore all the concerned models

* Remove print

* Update template

* Apply Sylvain's and Morgan's comments

* Fix cast

* Put the cast inside call

* Remove cond in ebds

* Fix funnel

* Restore previous dot product (attention_scores) computation

* Add ConvBERT and BART

* Make all the S2S models ONNX compliant

* Fix test

* Fix check copies
    31563e05
test_modeling_tf_common.py 57.5 KB