transformers/tests
Younes Belkada dfc76018c1
OPT-fix (#17229)
* try fixes

* Revert "try fixes"

This reverts commit a8ad75ef69.

* add correct shape

* add correct path
2022-05-13 15:14:23 +02:00
..
benchmark [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
deepspeed missing file (#17164) 2022-05-10 10:19:50 -07:00
extended Black preview (#17217) 2022-05-12 16:25:55 -04:00
fixtures add a warning in SpmConverter for sentencepiece's model using the byte fallback feature (#16629) 2022-04-11 11:06:10 +02:00
generation Black preview (#17217) 2022-05-12 16:25:55 -04:00
models OPT-fix (#17229) 2022-05-13 15:14:23 +02:00
onnx Black preview (#17217) 2022-05-12 16:25:55 -04:00
optimization [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
pipelines Black preview (#17217) 2022-05-12 16:25:55 -04:00
sagemaker Black preview (#17217) 2022-05-12 16:25:55 -04:00
tokenization [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
trainer Ensure tensors are at least 1d for pad and concat (#17179) 2022-05-11 13:19:08 -04:00
utils Black preview (#17217) 2022-05-12 16:25:55 -04: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 Black preview (#17217) 2022-05-12 16:25:55 -04:00
test_feature_extraction_common.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_common.py Black preview (#17217) 2022-05-12 16:25:55 -04:00
test_modeling_flax_common.py [Flax] improve large model init and loading (#16148) 2022-04-19 14:19:55 +02:00
test_modeling_tf_common.py Black preview (#17217) 2022-05-12 16:25:55 -04:00
test_sequence_feature_extraction_common.py Some tests misusing assertTrue for comparisons fix (#16771) 2022-04-19 14:44:08 +02:00
test_tokenization_common.py Black preview (#17217) 2022-05-12 16:25:55 -04:00