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	Rebase ESM PR and update all file formats (#19055)	2022-09-30 14:16:25 +01:00
test_modeling_esm.py	added support for gradient checkpointing in ESM models (#26386)	2023-09-26 10:15:53 +02:00
test_modeling_esmfold.py	CI with `num_hidden_layers=2` 🚀🚀🚀 (#25266)	2023-08-02 20:22:36 +02:00
test_modeling_tf_esm.py	Add tf_keras imports to prepare for Keras 3 (#28588)	2024-01-30 17:26:36 +00:00
test_tokenization_esm.py	Allow add_tokens for ESM (#28535)	2024-01-19 12:32:05 +00:00