PR #19321 Call _set_save_spec() when creating TF models

Call _set_save_spec() when creating TF models #19321

Rocketknight1 merged 8 commits into main from set_serving_spec_on_build

Add a build_from_serving_sig_and_dummies method and replace all calls…

4c29326f

make fixup

45c9862d

Remove the overridden save() as this is no longer necessary

d2fb3ae7

Rocketknight1 marked this pull request as ready for review 3 years ago

Rocketknight1 requested a review from

sgugger 3 years ago

Rocketknight1 requested a review from

gante 3 years ago

sgugger approved these changes on 2022-10-04

Also call _set_save_spec(), the last missing piece

0c011b77

Ensure we set the save spec when loading from config too

fbcaec85

Turn this whole thing into a one-line PR

decb608b

Turn this whole thing into a one-line PR

b2b25f19

Turn this whole thing into a one-line PR

cab014c0

Rocketknight1 changed the title ~~Add a build_from_serving_sig_and_dummies method~~ Call _set_save_spec() when creating TF models 3 years ago

Rocketknight1 merged 071df6eb into main 3 years ago

Rocketknight1 deleted the set_serving_spec_on_build branch 3 years ago

Reviewers

sgugger

gante

Assignees

No one assigned

Labels

None yet

Milestone

No milestone