transformers/tests
2023-01-31 11:33:18 +00:00
..
benchmark
deepspeed [examples/deepspeed] fix renamed api (#21283) 2023-01-24 09:54:33 -08:00
extended [bnb optim] fixing test (#21030) 2023-01-12 08:52:54 -08:00
fixtures
generation Template for framework-agnostic tests (#21348) 2023-01-31 11:33:18 +00:00
mixed_int8 [t5] Fix T5 inference in float16 + bnb error (#21281) 2023-01-24 18:14:38 +01:00
models Add DETA (#20983) 2023-01-31 10:43:10 +01:00
onnx Add Onnx Config for PoolFormer (#20868) 2022-12-23 01:30:57 -05:00
optimization
pipelines Pipeline testing - using tiny models on Hub (#20426) 2023-01-30 10:39:43 +01:00
repo_utils
sagemaker
tokenization
trainer Add AWS Neuron torchrun support (#20806) 2023-01-18 11:21:19 -05:00
utils Accept batched tensor of images as input to image processor (#21144) 2023-01-26 10:15:26 +00:00
__init__.py
test_configuration_common.py
test_feature_extraction_common.py Add test_image_processing_common.py (#20785) 2023-01-23 13:48:30 +00:00
test_image_processing_common.py Add test_image_processing_common.py (#20785) 2023-01-23 13:48:30 +00:00
test_image_transforms.py Move convert_to_rgb to image_transforms module (#20784) 2022-12-15 18:47:04 +00:00
test_modeling_common.py Use model_class.__name__ and compare against XXX_MAPPING_NAMES (#21304) 2023-01-26 11:31:31 +01:00
test_modeling_flax_common.py Generate: save generation config with the models' .save_pretrained() (#21264) 2023-01-23 16:21:44 +00:00
test_modeling_tf_common.py Generate: save generation config with the models' .save_pretrained() (#21264) 2023-01-23 16:21:44 +00:00
test_sequence_feature_extraction_common.py
test_tokenization_common.py