transformers/tests/utils
Lysandre Debut 113ebf80ac
Safetensors serialization by default (#27064)
* Safetensors serialization by default

* First pass on the tests

* Second pass on the tests

* Third pass on the tests

* Fix TF weight loading from TF-format safetensors

* Specific encoder-decoder fixes for weight crossloading

* Add VisionEncoderDecoder fixes for TF too

* Change filename test for pt-to-tf

* One missing fix for TFVisionEncoderDecoder

* Fix the other crossload test

* Support for flax + updated tests

* Apply suggestions from code review

Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>

* Sanchit's comments

* Sanchit's comments 2

* Nico's comments

* Fix tests

* cleanup

* Apply suggestions from code review

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

---------

Co-authored-by: Matt <rocketknight1@gmail.com>
Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
2023-10-31 19:16:49 +01:00
..
__init__.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_activations_tf.py TF: Add sigmoid activation function (#16819) 2022-04-19 16:13:08 +01:00
test_activations.py Add the GeLU activation from pytorch with the tanh approximation (#21345) 2023-02-02 09:33:04 -05:00
test_add_new_model_like.py Migrate doc files to Markdown. (#24376) 2023-06-20 18:07:47 -04:00
test_audio_utils.py Porting the torchaudio kaldi fbank implementation to audio_utils (#26182) 2023-09-21 17:52:47 +02:00
test_backbone_utils.py Add methods to update and verify out_features out_indices (#23031) 2023-05-04 10:15:06 +01:00
test_cli.py Safetensors serialization by default (#27064) 2023-10-31 19:16:49 +01:00
test_convert_slow_tokenizer.py Revert error back into warning for byte fallback conversion. (#22607) 2023-04-06 14:00:29 +02:00
test_doc_samples.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_dynamic_module_utils.py Fix the regex in get_imports to support multiline try blocks and excepts with specific exception types (#23725) 2023-05-24 15:40:19 -04:00
test_file_utils.py Inheritance-based framework detection (#21784) 2023-02-27 15:31:55 +00:00
test_generic.py Refactor conversion function (#19799) 2022-10-24 13:48:40 -04:00
test_hf_argparser.py ⚠️ Time to say goodbye to py37 (#24091) 2023-06-28 07:22:39 +02:00
test_hub_utils.py Fix gated repo tests (#26257) 2023-09-19 13:25:12 +02:00
test_image_processing_utils.py Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
test_image_utils.py Fix test_load_img_url_timeout (#25976) 2023-09-05 11:34:28 +02:00
test_logging.py Fix flaky test for log level (#21776) 2023-02-28 16:24:14 -05:00
test_model_card.py Black preview (#17217) 2022-05-12 16:25:55 -04:00
test_model_output.py Register ModelOutput as supported torch pytree nodes (#26618) 2023-10-24 11:02:40 +02:00
test_modeling_tf_core.py Make (TF) CI faster (test only a subset of model classes) (#24592) 2023-06-30 16:54:54 +02:00
test_offline.py Fix the expected error in test_offline_mode_pipeline_exception (#23022) 2023-04-27 14:22:05 +02:00
test_skip_decorators.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_versions_utils.py ⚠️ Time to say goodbye to py37 (#24091) 2023-06-28 07:22:39 +02:00
tiny_model_summary.json Fix failing MusicgenTest .test_pipeline_text_to_audio (#26586) 2023-10-06 15:53:59 +02:00