transformers/tests/models/xglm
Matt 3bd1fe4315
Stop storing references to bound methods via tf.function (#24146)
* Stop storing references to bound methods in tf.functions

* Remove the gc.collect calls now that we resolved the underlying problem

* Remove the default signature from model.serving entirely, big cleanup

* Remove _prune_signature as self.input_signature can prune itself

* Restore serving docstring

* Update int support test to check the input signature

* Make sure other tests also use model.input_signature and not serving.input_signature

* Restore _prune_signature

* Remove the doctest GC now it's no longer needed

* Correct core tests to use the pruned sig

* order lines correctly in core tests

* Add eager_serving back with a deprecation warning
2023-06-13 19:04:22 +01:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_flax_xglm.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_modeling_tf_xglm.py Stop storing references to bound methods via tf.function (#24146) 2023-06-13 19:04:22 +01:00
test_modeling_xglm.py Fix XGLM OOM on CI (#24123) 2023-06-09 15:20:59 +02:00
test_tokenization_xglm.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00