Matt 415e9a0980 Add tf_keras imports to prepare for Keras 3 (#28588) ... * Port core files + ESM (because ESM code is odd) * Search-replace in modelling code * Fix up transfo_xl as well * Fix other core files + tests (still need to add correct import to tests) * Fix cookiecutter * make fixup, fix imports in some more core files * Auto-add imports to tests * Cleanup, add imports to sagemaker tests * Use correct exception for importing tf_keras * Fixes in modeling_tf_utils * make fixup * Correct version parsing code * Ensure the pipeline tests correctly revert to float32 after each test * Ensure the pipeline tests correctly revert to float32 after each test * More tf.keras -> keras * Add dtype cast * Better imports of tf_keras * Add a cast for tf.assign, just in case * Fix callback imports		2024-01-30 17:26:36 +00:00
..
__init__.py	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
test_modeling_flax_gpt2.py	Correct attention mask dtype for Flax GPT2 (#25636)	2023-08-25 17:36:37 +02:00
test_modeling_gpt2.py	[`Styling`] stylify using ruff (#27144)	2023-11-16 17:43:19 +01:00
test_modeling_tf_gpt2.py	More TF fixes (#28081)	2023-12-18 15:26:03 +00:00
test_tokenization_gpt2.py	🚨🚨 🚨🚨 [`Tokenizer`] attemp to fix add_token issues🚨🚨 🚨🚨 (#23909)	2023-09-18 20:28:36 +02:00
test_tokenization_gpt2_tf.py	Add tf_keras imports to prepare for Keras 3 (#28588)	2024-01-30 17:26:36 +00:00