{
  "architectures": [
    "SegmentBorzoi"
  ],
  "attention_dim_key": 64,
  "auto_map": {
    "AutoConfig": "segment_borzoi.SegmentBorzoiConfig",
    "AutoModel": "segment_borzoi.SegmentBorzoi"
  },
  "dim_divisible_by": 32,
  "embed_dim": 1536,
  "features": [
    "protein_coding_gene",
    "lncRNA",
    "exon",
    "intron",
    "splice_donor",
    "splice_acceptor",
    "5UTR",
    "3UTR",
    "CTCF-bound",
    "polyA_signal",
    "enhancer_Tissue_specific",
    "enhancer_Tissue_invariant",
    "promoter_Tissue_specific",
    "promoter_Tissue_invariant"
  ],
  "model_type": "segment_borzoi",
  "num_attention_heads": 8,
  "num_rel_pos_features": 32,
  "torch_dtype": "float32",
  "transformers_version": "4.41.1"
}