transformers/tests
Younes Belkada f7ea959b96
[core/ GC / tests] Stronger GC tests (#27124)
* stronger GC tests

* better tests and skip failing tests

* break down into 3 sub-tests

* break down into 3 sub-tests

* refactor a bit

* more refactor

* fix

* last nit

* credits contrib and suggestions

* credits contrib and suggestions

---------

Co-authored-by: Yih-Dar <2521628+ydshieh@users.noreply.github.com>
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
2023-10-30 19:53:46 +01:00
..
benchmark
bettertransformer Fixed malapropism error (#26660) 2023-10-09 11:04:57 +02:00
deepspeed fix the deepspeed tests (#26021) 2023-09-13 10:26:53 +05:30
extended Device agnostic trainer testing (#27131) 2023-10-30 18:16:40 +00:00
fixtures
fsdp Skip TrainerIntegrationFSDP::test_basic_run_with_cpu_offload if torch < 2.1 (#26764) 2023-10-12 18:22:09 +02:00
generation Add early stopping for Bark generation via logits processor (#26675) 2023-10-27 11:07:33 +01:00
models [core/ GC / tests] Stronger GC tests (#27124) 2023-10-30 19:53:46 +01:00
optimization
peft_integration [PEFT] Final fixes (#26559) 2023-10-03 14:53:09 +02:00
pipelines Fix and re-enable ConversationalPipeline tests (#26907) 2023-10-19 12:04:25 +01:00
quantization [tests / Quantization] Fix bnb test (#27145) 2023-10-30 15:43:08 +01:00
repo_utils Docstring check (#26052) 2023-10-04 15:13:37 +02:00
sagemaker Add many missing spaces in adjacent strings (#26751) 2023-10-12 10:28:40 +02:00
tokenization [Tokenizer] Fix slow and fast serialization (#26570) 2023-10-18 16:30:53 +02:00
tools
trainer Device agnostic trainer testing (#27131) 2023-10-30 18:16:40 +00:00
utils Register ModelOutput as supported torch pytree nodes (#26618) 2023-10-24 11:02:40 +02:00
__init__.py
test_backbone_common.py [AutoBackbone] Add test (#26094) 2023-09-18 23:47:54 +02:00
test_configuration_common.py Deal with nested configs better in base class (#25237) 2023-08-04 14:56:09 +02:00
test_configuration_utils.py Deal with nested configs better in base class (#25237) 2023-08-04 14:56:09 +02:00
test_feature_extraction_common.py
test_feature_extraction_utils.py
test_image_processing_common.py Input data format (#25464) 2023-08-16 17:45:02 +01:00
test_image_processing_utils.py
test_image_transforms.py Add input_data_format argument, image transforms (#25462) 2023-08-11 15:09:31 +01:00
test_modeling_common.py [core/ GC / tests] Stronger GC tests (#27124) 2023-10-30 19:53:46 +01:00
test_modeling_flax_common.py
test_modeling_flax_utils.py
test_modeling_tf_common.py Skip test_onnx_runtime_optimize for now (#25560) 2023-08-17 11:23:16 +02:00
test_modeling_tf_utils.py
test_modeling_utils.py [Attention Mask] Refactor all encoder-decoder attention mask (#27086) 2023-10-27 16:42:01 +02:00
test_pipeline_mixin.py Fix and re-enable ConversationalPipeline tests (#26907) 2023-10-19 12:04:25 +01:00
test_sequence_feature_extraction_common.py Fix typo (#25966) 2023-09-05 10:12:25 +02:00
test_tokenization_common.py [Tokenizer] Fix slow and fast serialization (#26570) 2023-10-18 16:30:53 +02:00
test_tokenization_utils.py