| { | |
| "_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 | |
| } | |