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
..
benchmark	[Test refactor 1/5] Per-folder tests reorganization (#15725)	2022-02-23 15:46:28 -05:00
bettertransformer	Add methods to PreTrainedModel to use PyTorch's BetterTransformer (#21259)	2023-04-27 11:03:42 +02:00
bnb	[tests] fix bitsandbytes import issue (#24151)	2023-06-09 21:53:11 -07:00
deepspeed	accelerate deepspeed and gradient accumulation integrate (#23236)	2023-05-31 15:16:22 +05:30
extended	[tests] switch to torchrun (#22712)	2023-04-12 08:25:45 -07:00
fixtures	[WIP] add SpeechT5 model (#18922)	2023-02-03 12:43:46 -05:00
generation	Generate: GenerationConfig can overwrite attributes at from_pretrained time (#24238)	2023-06-13 17:59:21 +01:00
models	Stop storing references to bound methods via tf.function (#24146)	2023-06-13 19:04:22 +01:00
onnx	Fix issue introduced in PR #23163 (#23363)	2023-05-15 11:38:44 +02:00
optimization	Make schedulers picklable by making lr_lambda fns global (#21768)	2023-03-02 12:08:43 -05:00
pipelines	Fix Pipeline CI OOM issue (#24124)	2023-06-09 16:49:02 +02:00
repo_utils	Fix expected value in tests of the test fetcher (#24077)	2023-06-07 11:38:56 -04:00
sagemaker	Avoid invalid escape sequences, use raw strings (#22936)	2023-04-25 09:17:56 -04:00
tokenization	Update quality tooling for formatting (#21480)	2023-02-06 18:10:56 -05:00
tools	Tool types (#24032)	2023-06-09 13:34:07 -04:00
trainer	🚨🚨🚨 Replace DataLoader logic for Accelerate in Trainer, remove unneeded tests 🚨🚨🚨 (#24028)	2023-06-12 11:23:37 -04:00
utils	Stop storing references to bound methods via tf.function (#24146)	2023-06-13 19:04:22 +01:00
__init__.py	GPU text generation: mMoved the encoded_prompt to correct device	2020-01-06 15:11:12 +01:00
test_backbone_common.py	Add TimmBackbone model (#22619)	2023-06-06 17:11:30 +01:00
test_configuration_common.py	Add RWKV-4 (#22797)	2023-05-09 13:04:10 -04:00
test_feature_extraction_common.py	Prepare tests for hfh 0.14 (#22958)	2023-04-24 09:31:50 -04:00
test_image_processing_common.py	Prepare tests for hfh 0.14 (#22958)	2023-04-24 09:31:50 -04:00
test_image_transforms.py	Bug fix - flip_channel_order for channels first images (#23701)	2023-05-31 17:12:27 +01:00
test_modeling_common.py	Tied params cleanup (#24211)	2023-06-13 11:38:39 -04:00
test_modeling_flax_common.py	Fix bigbird random attention (#21023)	2023-04-27 13:52:28 -04:00
test_modeling_tf_common.py	Stop storing references to bound methods via tf.function (#24146)	2023-06-13 19:04:22 +01:00
test_pipeline_mixin.py	Update tiny models and pipeline tests (#23446)	2023-05-18 17:29:04 +02:00
test_sequence_feature_extraction_common.py	Apply ruff flake8-comprehensions (#21694)	2023-02-22 09:14:54 +01:00
test_tokenization_common.py	chore: allow protobuf 3.20.3 requirement (#22759)	2023-05-10 20:22:56 +02:00