{ "alphabet_size": { "methylation": 66, "rnaseq": 66 }, "architectures": [ "MOJO" ], "attention_maps_to_save": [], "auto_map": { "AutoConfig": "mojo.MOJOConfig", "AutoModel": "mojo.MOJO" }, "conv_init_embed_dim": 512, "embed_dim": 512, "embeddings_layers_to_save": [], "ffn_embed_dim": 1024, "filter_list": [ 512, 512, 512, 512, 512, 512, 512, 512, 512 ], "fixed_sequence_length": 17152, "init_gene_embed_dim": 200, "key_size": 32, "model_type": "MOJO", "num_attention_heads": 16, "num_downsamples": 8, "num_hidden_layers_head": 1, "num_layers": 8, "project_gene_embedding": true, "sequence_length": 17116, "stem_kernel_shape": 15, "token_embed_dim": 256, "torch_dtype": "float32", "transformers_version": "4.37.2", "use_gene_embedding": true }