| { |
| "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.51.0", |
| "use_gene_embedding": true |
| } |
|
|
