transformers/tests
Stas Bekman 2f5507580b
[from_pretrained] extend torch_dtype="auto" to look up config.torch_dtype first, expand docs (#21524)
* [from_pretrained] expand on torch_dtype entry

* fold 4 into 1

* style

* support torch_dtype='config' plus tests

* style

* oops

* fold config into auto, fix bug

* fix check

* better log

* better log

* clean up
2023-02-10 09:09:21 -08: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 Add X-MOD (#20939) 2023-02-10 15:32:06 +01: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