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_data_collator.py	handle numpy inputs in whole word mask data collator (#22032)	2023-03-10 10:50:29 -05:00
test_trainer_callback.py	Apply ruff flake8-comprehensions (#21694)	2023-02-22 09:14:54 +01:00
test_trainer_distributed.py	Patch with accelerate xpu (#25714)	2023-09-05 15:41:42 +01:00
test_trainer_seq2seq.py	Fix TypeError: Object of type int64 is not JSON serializable (#24340)	2023-06-27 12:15:49 +01:00
test_trainer_tpu.py	[Test refactor 1/5] Per-folder tests reorganization (#15725)	2022-02-23 15:46:28 -05:00
test_trainer_utils.py	Apply ruff flake8-comprehensions (#21694)	2023-02-22 09:14:54 +01:00
test_trainer.py	Safetensors serialization by default (#27064)	2023-10-31 19:16:49 +01:00