transformers
Stop storing references to bound methods via tf.function
#24146
Merged
Merged
Stop storing references to bound methods via tf.function #24146
Rocketknight1
merged
12 commits into
main
from
fix_tf_model_garbage_collection
ydshieh
approved these changes
on 2023-06-13
Stop storing references to bound methods in tf.functions
8da21872
Remove the gc.collect calls now that we resolved the underlying problem
5bb70c12
Remove the default signature from model.serving entirely, big cleanup
c5466d6a
Remove _prune_signature as self.input_signature can prune itself
03c90611
Restore serving docstring
d8c32e4b
Update int support test to check the input signature
0bc46e06
Make sure other tests also use model.input_signature and not serving.…
a68b2ac9
Restore _prune_signature
e371f62e
Remove the doctest GC now it's no longer needed
f98b7b74
Correct core tests to use the pruned sig
8af7857c
order lines correctly in core tests
5939c126
gante
approved these changes
on 2023-06-13
amyeroberts
commented
on 2023-06-13
Add eager_serving back with a deprecation warning
8764c371
amyeroberts
approved these changes
on 2023-06-13
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub