{
  "_class_name": "RFD4ProteinaTransformer",
  "_diffusers_version": "0.42.0.dev0",
  "cath_emb_dim": 32,
  "dim_cond": 256,
  "feats_cond_seq": [
    "time_emb_bb_ca",
    "time_emb_local_latents",
    "cath_code",
    "recycle_remaining"
  ],
  "feats_pair_cond": [
    "time_emb_bb_ca",
    "time_emb_local_latents"
  ],
  "feats_pair_repr": [
    "xt_bb_ca_pair_dists",
    "bond_mask",
    "bond_order",
    "is_aromatic",
    "index_group_pair",
    "molecule_idx_pair",
    "molecule_entity_pair",
    "token_bond_path_length",
    "canonical_polymer_ordering",
    "na_ss_pair"
  ],
  "feats_seq": [
    "xt_bb_ca",
    "xt_local_latents",
    "token_type",
    "sequence_probabilities",
    "cond_a37coors_nm",
    "cond_a37coors_internal_nm",
    "rasa_binned",
    "secondary_structure",
    "hbond_modernized",
    "pistack",
    "pication",
    "saltbridge",
    "hydrophobic",
    "metal_coord",
    "na_ss_state",
    "hotspot",
    "atom_charge",
    "atomic_element",
    "coord_group_seq",
    "chirality",
    "chain_centroid_xyz"
  ],
  "latent_dim": 32,
  "nheads": 12,
  "nlayers": 14,
  "num_sequence_classes": 147,
  "num_token_types": 7,
  "pair_repr_dim": 384,
  "recycle_conditioning_dim": 1,
  "seq_sep_dim": 64,
  "t_emb_dim": 256,
  "token_dim": 768,
  "token_type_emb_dim": 32,
  "tri_mult_c": 256,
  "update_pair_repr_every_n": 2,
  "xt_pair_dist_dim": 30,
  "xt_pair_dist_max": 3,
  "xt_pair_dist_min": 0.1
}