{"name": "model_4", "trainable": true, "layers": [{"module": "keras.layers", "class_name": "InputLayer", "config": {"batch_input_shape": [null, 8192], "dtype": "int32", "sparse": false, "ragged": false, "name": "input_5"}, "registered_name": null, "name": "input_5", "inbound_nodes": []}, {"module": "tokun.model", "class_name": "Encoder", "config": {"token_dim": [4, 4], "encoding_dim": 256, "embedding_dim": 256, "sequence_axis": 1, "feature_axis": -1, "activation": "gelu"}, "registered_name": "models>Encoder", "build_config": {"input_shape": [16, 2048]}, "name": "encoder_4", "inbound_nodes": [[["input_5", 0, 0, {}]]]}, {"module": "tokun.model", "class_name": "Decoder", "config": {"token_dim": [4, 4], "encoding_dim": 8, "embedding_dim": 256, "sequence_axis": 1, "feature_axis": -1, "activation": "gelu", "output": "binary"}, "registered_name": "models>Decoder", "build_config": {"input_shape": [16, 128, 256]}, "name": "decoder_4", "inbound_nodes": [[["encoder_4", 0, 0, {}]]]}], "input_layers": [["input_5", 0, 0]], "output_layers": [["decoder_4", 0, 0]]}