Export Gemma 2 from Keras to Saved Model?

Ok, downloaded Gemma2 from
https://www.kaggle.com/models/google/gemma-2/keras

-unpacked and got the following files:

/assets
config.json
metadata.json
model.weights.h5
tokenizer.json

-tried to load model by using following Python script:

import json
from tensorflow import keras

# 1. Path to the config and weights:
config_path = "d:/Install/TensorFlow/models/Gemma_2/config.json"
weights_path = "d:/Install/TensorFlow/models/Gemma_2/model.weights.h5"
saved_model_dir = "d:/Install/TensorFlow/models/Gemma_2/gemma2_saved_model"

# 2. load model configuration
with open(config_path, 'r') as f:
    model_config = json.load(f)

# 3. Reconstruct the model:
model = keras.models.model_from_json(json.dumps(model_config))

# 4. Load the weights:
model.load_weights(weights_path)

# 5. Finally, save model as  TensorFlow SavedModel format:
model.save(saved_model_dir, save_format="tf")

print("Model saved to the :", saved_model_dir)

-at the line

model = keras.models.model_from_json(json.dumps(model_config))

got following error:

ValueError: Unknown layer: GemmaBackbone. Please ensure this object is passed to the custom_objectsargument. See https://www.tensorflow.org/guide/keras/save_and_serialize#registering_the_custom_object for details.

Questions:

• Can someone confirm that I'm on the right track?
• Looked at
• https://www.tensorflow.org/guide/keras/save_and_serialize#registering_the_custom_object

but not sure what to do with GemmaBackbone...huh?

any help?

Bắt đầu với script Python, đặc biệt là lệnh gọi model_from_json bị lỗi trên GemmaBackbone, và xem xét các tệp config.json, model.weights.h5 và tokenizer.json đã tải xuống. Xác định xem repository này có hỗ trợ xuất gói Gemma 2 này sang SavedModel hay không; hoàn tất có nghĩa là ghi lại một quy trình được hỗ trợ hoặc giải thích rõ ràng rằng tính năng này không được hỗ trợ.