| {"name": "model_2", "trainable": true, "layers": [{"module": "keras.layers", "class_name": "InputLayer", "config": {"batch_input_shape": [null, 32768], "dtype": "int32", "sparse": false, "ragged": false, "name": "input_3"}, "registered_name": null, "name": "input_3", "inbound_nodes": []}, {"module": "tokun.model", "class_name": "Encoder", "config": {"token_dim": [16, 4], "encoding_dim": 256, "embedding_dim": 256, "latent_dim": 256, "activation": "gelu", "sequence_axis": 1, "feature_axis": -1}, "registered_name": "models>Encoder", "build_config": {"input_shape": [null, 1024]}, "name": "encoder_2", "inbound_nodes": [[["input_3", 0, 0, {}]]]}, {"module": "tokun.model", "class_name": "Decoder", "config": {"token_dim": [4, 16], "encoding_dim": 256, "embedding_dim": 256, "latent_dim": 256, "activation": "gelu", "sequence_axis": 1, "feature_axis": -1}, "registered_name": "models>Decoder", "build_config": {"input_shape": [null, 16, 256]}, "name": "decoder_2", "inbound_nodes": [[["encoder_2", 0, 0, {}]]]}], "input_layers": [["input_3", 0, 0]], "output_layers": [["decoder_2", 0, 0]]} |