config.json

{
    "module": "zea.models.hvae",
    "class_name": "HierarchicalVAE",
    "config": {
        "name": "hvae",
        "trainable": false,
        "dtype": {
            "module": "keras",
            "class_name": "DTypePolicy",
            "config": {
                "name": "float32"
            },
            "registered_name": null
        }
    },
    "registered_name": "HierarchicalVAE"
}