René's URL Explorer Experiment

{"@context":"https://schema.org","@type":"DiscussionForumPosting","headline":"Export Gemma 2 from Keras to Saved Model?","articleBody":"Ok, downloaded Gemma2 from\r\nhttps://www.kaggle.com/models/google/gemma-2/keras\r\n\r\n-unpacked and got the following files:\r\n```\r\n/assets\r\nconfig.json\r\nmetadata.json\r\nmodel.weights.h5\r\ntokenizer.json\r\n```\r\n-tried to load model by using following Python script:\r\n\r\n```\r\nimport json\r\nfrom tensorflow import keras\r\n\r\n# 1. Path to the config and weights:\r\nconfig_path = \"d:/Install/TensorFlow/models/Gemma_2/config.json\"\r\nweights_path = \"d:/Install/TensorFlow/models/Gemma_2/model.weights.h5\"\r\nsaved_model_dir = \"d:/Install/TensorFlow/models/Gemma_2/gemma2_saved_model\"\r\n\r\n# 2. load model configuration\r\nwith open(config_path, 'r') as f:\r\n    model_config = json.load(f)\r\n\r\n# 3. Reconstruct the model:\r\nmodel = keras.models.model_from_json(json.dumps(model_config))\r\n\r\n# 4. Load the weights:\r\nmodel.load_weights(weights_path)\r\n\r\n# 5. Finally, save model as  TensorFlow SavedModel format:\r\nmodel.save(saved_model_dir, save_format=\"tf\")\r\n\r\nprint(\"Model saved to the :\", saved_model_dir)\r\n```\r\n-at the line\r\n\r\n`model = keras.models.model_from_json(json.dumps(model_config))`\r\n\r\ngot following error:\r\n\r\n`ValueError: Unknown layer: GemmaBackbone. Please ensure this object is passed to the `custom_objects` argument. See https://www.tensorflow.org/guide/keras/save_and_serialize#registering_the_custom_object for details.\r\n`\r\n\r\nQuestions:\r\n\r\n-  Can someone confirm that I'm on the right track?\r\n-  Looked at\r\n-  https://www.tensorflow.org/guide/keras/save_and_serialize#registering_the_custom_object\r\n\r\n but not sure what to do with GemmaBackbone...huh?\r\n \r\n any help?\r\n\r\n\r\n\r\n","author":{"url":"https://github.com/JeeDevUser","@type":"Person","name":"JeeDevUser"},"datePublished":"2024-12-07T10:54:17.000Z","interactionStatistic":{"@type":"InteractionCounter","interactionType":"https://schema.org/CommentAction","userInteractionCount":7},"url":"https://github.com/595/java/issues/595"}