transformers/tests
amyeroberts cb56590111
Replace input_values_processing with unpack_inputs (#21502)
* Replace input_values_prrocessing with unpack_inputs

* Skip test failing with OOM

* Update tests
2023-02-10 18:19:39 +00:00
..
benchmark [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
deepspeed [tests] add missing report_to none (#21505) 2023-02-08 09:32:40 -08:00
extended Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
fixtures [WIP] add SpeechT5 model (#18922) 2023-02-03 12:43:46 -05:00
generation Generate: TF .generate() can now be exported with dynamic length (#21474) 2023-02-09 12:52:30 +00:00
mixed_int8 [bnb] Fine-tuning HF 8-bit models (#21290) 2023-02-02 16:39:23 +01:00
models Replace input_values_processing with unpack_inputs (#21502) 2023-02-10 18:19:39 +00:00
onnx Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
optimization Add inverse sqrt learning rate scheduler (#21495) 2023-02-07 15:00:50 -05:00
pipelines [pipeline] A simple fix for half-precision & 8bit models (#21479) 2023-02-10 10:26:17 +01:00
repo_utils Cleanup quality (#21493) 2023-02-07 12:27:31 -05:00
sagemaker Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
tokenization Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
trainer Fix epoch number when resuming training (#21478) 2023-02-06 19:34:34 -05:00
utils Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
__init__.py GPU text generation: mMoved the encoded_prompt to correct device 2020-01-06 15:11:12 +01:00
test_configuration_common.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_feature_extraction_common.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_image_processing_common.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_image_transforms.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_modeling_common.py [from_pretrained] extend torch_dtype="auto" to look up config.torch_dtype first, expand docs (#21524) 2023-02-10 09:09:21 -08:00
test_modeling_flax_common.py [Tests] Improve flax test_attention_outputs (#21486) 2023-02-10 11:31:49 -05:00
test_modeling_tf_common.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_sequence_feature_extraction_common.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_tokenization_common.py Cleanup quality (#21493) 2023-02-07 12:27:31 -05:00