Skip to content

[TensorFlow] Upgrade to keras v3Β #1461

Open
@felixdittrich92

Description

πŸš€ The feature

TensorFlow >= 2.16.0 switches to Keras v3 (keras-core) which will especially break the saving and loading logic from models in our case

  • pin to >=TF 2.16.0 (requires py3.9 min)

Keras: https://keras.io/api/models/model_saving_apis/weights_saving_and_loading/#save_weights-method
tf2onnx: https://github.com/onnx/tensorflow-onnx?tab=readme-ov-file#from_keras-tf-20-and-newer
hf hub: https://huggingface.co/docs/huggingface_hub/guides/upload

(Optional) Provide a script to convert old models to new format

Metadata

Type

No type

Projects

No projects

Relationships

None yet

Development

No branches or pull requests

Issue actions