Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Matt's avatar
    Update serving signatures and make sure we actually use them (#19034) · 2322eb8e
    Matt authored 
    * Override save() to use the serving signature as the default

* Replace int32 with int64 in all our serving signatures

* Remember one very important line so as not to break every test at once

* Dtype fix for TFLED

* dtype fix for shift_tokens_right in general

* Dtype fixes in mBART and RAG

* Fix dtypes for test_unpack_inputs

* More dtype fixes

* Yet more mBART + RAG dtype fixes

* Yet more mBART + RAG dtype fixes

* Add a check that the model actually has a serving method
    2322eb8e
test_modeling_tf_common.py 102 KB