config.json

{
    "module": "usbmd.models.taesd",
    "class_name": "TinyAutoencoder",
    "config": {
        "name": "tiny_autoencoder",
        "trainable": true,
        "dtype": {
            "module": "keras",
            "class_name": "DTypePolicy",
            "config": {
                "name": "float32"
            },
            "registered_name": null
        }
    },
    "registered_name": "TinyAutoencoder"
}