transformers/tests
Crystina 692e61e91a
Flax t5 Encoder (#17784)
* first draft adding Flax-t5-encoder and Flax-mt5-encoder

* imports

* after make fixup

* flax t5 encoder test

* black on test

* make fix-copies

* clean

* all_model_classes -> tuple

* clean test

* is_encoder_decoder=False in t5-enc tester

* remove file docstring before FlaxT5Encoder

* black

* isort

* commit suggestions on src/transformers/models/t5/modeling_flax_t5.py

Co-authored-by: Suraj Patil <surajp815@gmail.com>

* commit suggestions on src/transformers/models/t5/modeling_flax_t5.py

Co-authored-by: Suraj Patil <surajp815@gmail.com>

* Apply suggestions from code review

Co-authored-by: Suraj Patil <surajp815@gmail.com>

* remove _get_encoder_module

* self.decoder_seq_length -> self.encoder_seq_length as t5-enc does not have decoder

* bugfix - self.module_class is class itself, not instance;

* docs for mt5 and t5

* call -> __call__ in t5 doc

* FlaxMT5EncoderModel to TYPE_HINT

* run doc-builder to allow change the files

Co-authored-by: Suraj Patil <surajp815@gmail.com>
2022-06-30 00:49:02 +02:00
..
benchmark [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
deepspeed deprecate is_torch_bf16_available (#17738) 2022-06-20 08:40:11 -04:00
extended Update self-push workflow (#17177) 2022-05-13 16:28:00 +02: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 Update expected values in constrained beam search tests (#17887) 2022-06-28 08:53:53 +02:00
models Flax t5 Encoder (#17784) 2022-06-30 00:49:02 +02:00
onnx Mrbean/codegen onnx (#17903) 2022-06-28 14:57:53 +02:00
optimization [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
pipelines Fix img seg tests (load checkpoints from hf-internal-testing) (#17939) 2022-06-29 10:19:37 -04:00
sagemaker Black preview (#17217) 2022-05-12 16:25:55 -04:00
tokenization fix train_new_from_iterator in the case of byte-level tokenizers (#17549) 2022-06-08 15:30:41 +02:00
trainer Fix test_number_of_steps_in_training_with_ipex (#17889) 2022-06-28 08:55:02 +02:00
utils 🐛 Properly raise RepoNotFoundError when not authenticated (#17651) 2022-06-10 15:41:53 +02: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 Prepare transformers for v0.8.0 huggingface-hub release (#17716) 2022-06-21 11:51:18 -04:00
test_feature_extraction_common.py Compute min_resolution in prepare_image_inputs (#17915) 2022-06-29 10:30:20 +02:00
test_modeling_common.py Offload fixes (#17810) 2022-06-22 12:23:07 -04:00
test_modeling_flax_common.py Flax sharded (#17760) 2022-06-22 07:04:35 +02:00
test_modeling_tf_common.py TF: XLA beam search + most generation-compatible models are now also XLA-generate-compatible (#17857) 2022-06-29 12:41:01 +01: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 Fix properties of unset special tokens in non verbose mode (#17797) 2022-06-23 14:40:13 +02:00