Upload 2 files

model_5/config.json

@@ -0,0 +1,945 @@
+{
+  "CONFIG": {
+    "data": {
+      "transform": {
+        "common": {
+          "crop_size": {
+            "eval": null,
+            "finetune": 384,
+            "predict": null,
+            "train": 256
+          },
+          "feats": {
+            "aatype": [
+              "num residues placeholder"
+            ],
+            "all_atom_mask": [
+              "num residues placeholder",
+              null
+            ],
+            "all_atom_positions": [
+              "num residues placeholder",
+              null,
+              null
+            ],
+            "alt_chi_angles": [
+              "num residues placeholder",
+              null
+            ],
+            "assembly_num_chains": [
+              null
+            ],
+            "asym_id": [
+              "num residues placeholder"
+            ],
+            "asym_len": [
+              null
+            ],
+            "atom14_alt_gt_exists": [
+              "num residues placeholder",
+              null
+            ],
+            "atom14_alt_gt_positions": [
+              "num residues placeholder",
+              null,
+              null
+            ],
+            "atom14_atom_exists": [
+              "num residues placeholder",
+              null
+            ],
+            "atom14_atom_is_ambiguous": [
+              "num residues placeholder",
+              null
+            ],
+            "atom14_gt_exists": [
+              "num residues placeholder",
+              null
+            ],
+            "atom14_gt_positions": [
+              "num residues placeholder",
+              null,
+              null
+            ],
+            "atom37_atom_exists": [
+              "num residues placeholder",
+              null
+            ],
+            "backbone_rigid_mask": [
+              "num residues placeholder"
+            ],
+            "backbone_rigid_tensor": [
+              "num residues placeholder",
+              null,
+              null
+            ],
+            "bert_mask": [
+              "msa placeholder",
+              "num residues placeholder"
+            ],
+            "chi_angles_sin_cos": [
+              "num residues placeholder",
+              null,
+              null
+            ],
+            "chi_mask": [
+              "num residues placeholder",
+              null
+            ],
+            "cluster_bias_mask": [
+              "msa placeholder"
+            ],
+            "entity_id": [
+              "num residues placeholder"
+            ],
+            "extra_deletion_value": [
+              "extra msa placeholder",
+              "num residues placeholder"
+            ],
+            "extra_has_deletion": [
+              "extra msa placeholder",
+              "num residues placeholder"
+            ],
+            "extra_msa": [
+              "extra msa placeholder",
+              "num residues placeholder"
+            ],
+            "extra_msa_mask": [
+              "extra msa placeholder",
+              "num residues placeholder"
+            ],
+            "extra_msa_row_mask": [
+              "extra msa placeholder"
+            ],
+            "is_distillation": [],
+            "msa_feat": [
+              "msa placeholder",
+              "num residues placeholder",
+              null
+            ],
+            "msa_mask": [
+              "msa placeholder",
+              "num residues placeholder"
+            ],
+            "msa_row_mask": [
+              "msa placeholder"
+            ],
+            "no_recycling_iters": [],
+            "num_sym": [
+              "num residues placeholder"
+            ],
+            "pseudo_beta": [
+              "num residues placeholder",
+              null
+            ],
+            "pseudo_beta_mask": [
+              "num residues placeholder"
+            ],
+            "residue_index": [
+              "num residues placeholder"
+            ],
+            "residx_atom14_to_atom37": [
+              "num residues placeholder",
+              null
+            ],
+            "residx_atom37_to_atom14": [
+              "num residues placeholder",
+              null
+            ],
+            "resolution": [],
+            "rigidgroups_alt_gt_frames": [
+              "num residues placeholder",
+              null,
+              null,
+              null
+            ],
+            "rigidgroups_group_exists": [
+              "num residues placeholder",
+              null
+            ],
+            "rigidgroups_group_is_ambiguous": [
+              "num residues placeholder",
+              null
+            ],
+            "rigidgroups_gt_exists": [
+              "num residues placeholder",
+              null
+            ],
+            "rigidgroups_gt_frames": [
+              "num residues placeholder",
+              null,
+              null,
+              null
+            ],
+            "seq_length": [],
+            "seq_mask": [
+              "num residues placeholder"
+            ],
+            "sym_id": [
+              "num residues placeholder"
+            ],
+            "target_feat": [
+              "num residues placeholder",
+              null
+            ],
+            "template_aatype": [
+              "num templates placeholder",
+              "num residues placeholder"
+            ],
+            "template_all_atom_mask": [
+              "num templates placeholder",
+              "num residues placeholder",
+              null
+            ],
+            "template_all_atom_positions": [
+              "num templates placeholder",
+              "num residues placeholder",
+              null,
+              null
+            ],
+            "template_alt_torsion_angles_sin_cos": [
+              "num templates placeholder",
+              "num residues placeholder",
+              null,
+              null
+            ],
+            "template_backbone_rigid_mask": [
+              "num templates placeholder",
+              "num residues placeholder"
+            ],
+            "template_backbone_rigid_tensor": [
+              "num templates placeholder",
+              "num residues placeholder",
+              null,
+              null
+            ],
+            "template_mask": [
+              "num templates placeholder"
+            ],
+            "template_pseudo_beta": [
+              "num templates placeholder",
+              "num residues placeholder",
+              null
+            ],
+            "template_pseudo_beta_mask": [
+              "num templates placeholder",
+              "num residues placeholder"
+            ],
+            "template_sum_probs": [
+              "num templates placeholder",
+              null
+            ],
+            "template_torsion_angles_mask": [
+              "num templates placeholder",
+              "num residues placeholder",
+              null
+            ],
+            "template_torsion_angles_sin_cos": [
+              "num templates placeholder",
+              "num residues placeholder",
+              null,
+              null
+            ],
+            "true_msa": [
+              "msa placeholder",
+              "num residues placeholder"
+            ],
+            "use_clamped_fape": []
+          },
+          "max_extra_msa": {
+            "eval": 512,
+            "finetune": 5120,
+            "predict": 1024,
+            "train": 1024
+          },
+          "max_msa_clusters": {
+            "eval": 128,
+            "finetune": 512,
+            "predict": 512,
+            "train": 128
+          },
+          "max_templates": 4,
+          "no_recycling_iters": 3,
+          "reduce_msa_clusters_by_max_templates": false,
+          "resample_msa_in_recycling": true,
+          "subsample_templates": {
+            "eval": false,
+            "finetune": true,
+            "predict": false,
+            "train": true
+          },
+          "supervised": {
+            "eval": true,
+            "finetune": true,
+            "predict": false,
+            "train": true
+          },
+          "supervised_features": [
+            "all_atom_mask",
+            "all_atom_positions",
+            "resolution",
+            "use_clamped_fape",
+            "is_distillation"
+          ],
+          "template_features": [
+            "template_all_atom_positions",
+            "template_sum_probs",
+            "template_aatype",
+            "template_all_atom_mask"
+          ],
+          "uniform_recycling": {
+            "eval": false,
+            "finetune": true,
+            "predict": false,
+            "train": true
+          },
+          "unsupervised_features": [
+            "aatype",
+            "residue_index",
+            "msa",
+            "num_alignments",
+            "seq_length",
+            "between_segment_residues",
+            "deletion_matrix",
+            "no_recycling_iters"
+          ]
+        },
+        "make_masked_msa": {
+          "profile_prob": 0.1,
+          "replace_fraction": 0.15,
+          "same_prob": 0.1,
+          "uniform_prob": 0.1
+        },
+        "make_relative_encoding_feat": {
+          "max_relative_feature": 32
+        },
+        "nearest_neighbor_clusters": {
+          "gap_agreement_weight": 0.0
+        },
+        "randomly_replace_msa_with_unknown": {
+          "replace_proportion": 0.0
+        },
+        "sample_msa": {
+          "reduce_msa_clusters_by_max_templates": false
+        }
+      }
+    },
+    "globals": {
+      "is_multimer": false,
+      "mode": "predict",
+      "num_recycles": 3,
+      "recycle_early_stop_tolerance": 0.5,
+      "use_templates": false
+    },
+    "model": {
+      "auxiliary_heads": {
+        "c_m": 256,
+        "c_s": 384,
+        "c_z": 128,
+        "distogram_head_num_bins": 64,
+        "experimentally_resolved_head_c_out": 37,
+        "masked_msa_head_c_out": 23,
+        "predicted_aligned_error_head_num_bins": 64,
+        "predicted_lddt_head_c_hidden": 128,
+        "predicted_lddt_head_num_bins": 50,
+        "tm_enabled": false
+      },
+      "evoformer_stack": {
+        "c_h_m": 32,
+        "c_h_opm": 32,
+        "c_h_u": 128,
+        "c_h_z": 32,
+        "c_m": 256,
+        "c_s": 384,
+        "c_z": 128,
+        "fused": false,
+        "h_m": 8,
+        "h_z": 4,
+        "msa_global_col_attention": false,
+        "n_trans": 4,
+        "num_blocks": 48,
+        "opm_first": false,
+        "ouput_single": true
+      },
+      "extra_msa_stack": {
+        "c_h_m": 8,
+        "c_h_opm": 32,
+        "c_h_u": 128,
+        "c_h_z": 32,
+        "c_m": 64,
+        "c_z": 128,
+        "fused": false,
+        "h_m": 8,
+        "h_z": 4,
+        "msa_global_col_attention": true,
+        "n_trans": 4,
+        "num_blocks": 4,
+        "opm_first": false
+      },
+      "feature_embedder": {
+        "c_e": 64,
+        "c_m": 256,
+        "c_t": 64,
+        "c_taf": 57,
+        "c_tpf": 88,
+        "c_z": 128,
+        "extra_msa_dim": 25,
+        "max_relative_feature": 32,
+        "msa_dim": 49,
+        "num_bins": 15,
+        "tf_dim": 22
+      },
+      "loss": {
+        "distogram_loss": {
+          "eps": 1e-06,
+          "max_bin": 21.6875,
+          "min_bin": 2.3125,
+          "no_bins": 64,
+          "weight": 0.3
+        },
+        "experimentally_resolved_loss": {
+          "eps": 1e-08,
+          "max_resolution": 3.0,
+          "min_resolution": 0.1,
+          "weight": 0.0
+        },
+        "fape_loss": {
+          "backbone": {
+            "clamp_distance": 10.0,
+            "loss_unit_distance": 10.0,
+            "weight": 0.5
+          },
+          "eps": 0.0001,
+          "sidechain": {
+            "clamp_distance": 10.0,
+            "length_scale": 10.0,
+            "weight": 0.5
+          },
+          "weight": 1.0
+        },
+        "masked_msa_loss": {
+          "eps": 1e-08,
+          "weight": 2.0
+        },
+        "predicted_lddt_loss": {
+          "cutoff": 15.0,
+          "eps": 1e-10,
+          "max_resolution": 3.0,
+          "min_resolution": 0.1,
+          "no_bins": 50,
+          "weight": 0.01
+        },
+        "predicted_tm_loss": {
+          "eps": 1e-08,
+          "max_bin": 31,
+          "max_resolution": 3.0,
+          "min_resolution": 0.1,
+          "no_bins": 64,
+          "weight": 0.1
+        },
+        "supervised_chi_loss": {
+          "angle_norm_weight": 0.01,
+          "chi_weight": 0.5,
+          "eps": 1e-06,
+          "weight": 1.0
+        },
+        "violation_loss": {
+          "clash_overlap_tolerance": 1.5,
+          "eps": 1e-06,
+          "violation_tolerance_factor": 12.0,
+          "weight": 0.0
+        }
+      },
+      "structure_module": {
+        "c_s": 384,
+        "c_sc": 128,
+        "c_z": 128,
+        "dropout": 0.1,
+        "ipa_scalar_bias": true,
+        "no_angles": 7,
+        "num_head": 12,
+        "num_layer": 8,
+        "num_point_qk": 4,
+        "num_point_v": 8,
+        "num_scalar_qk": 16,
+        "num_scalar_v": 16,
+        "position_scale": 10.0,
+        "separate_kv": false
+      },
+      "template_pair_stack": {
+        "attention_first": true,
+        "c_h_t": 16,
+        "c_h_u": 64,
+        "c_t": 64,
+        "c_z": 128,
+        "fused": false,
+        "h_t": 4,
+        "n_trans": 2,
+        "num_blocks": 2
+      }
+    }
+  },
+  "CONFIG_DIFFS": {
+    "model_1": {
+      "data.transform.common.max_extra_msa.predict": 5120,
+      "data.transform.common.max_extra_msa.train": 5120,
+      "data.transform.use_templates": true,
+      "globals.use_templates": true
+    },
+    "model_1_multimer": {
+      "data.transform.common.max_extra_msa": {
+        "eval": 1152,
+        "finetune": 1152,
+        "predict": 1152,
+        "train": 1152
+      },
+      "data.transform.common.max_msa_clusters": {
+        "eval": 128,
+        "finetune": 252,
+        "predict": 128,
+        "train": 128
+      },
+      "globals.is_multimer": true,
+      "globals.use_templates": true,
+      "model.auxiliary_heads.masked_msa_head_c_out": 22,
+      "model.auxiliary_heads.tm_enabled": true,
+      "model.evoformer_stack.opm_first": true,
+      "model.extra_msa_stack.opm_first": true,
+      "model.feature_embedder.c_taf": 34,
+      "model.feature_embedder.tf_dim": 21,
+      "model.loss.supervised_chi_loss.weight": 3.0,
+      "model.structure_module.ipa_scalar_bias": false,
+      "model.structure_module.position_scale": 20,
+      "model.structure_module.separate_kv": true,
+      "model.template_pair_stack.attention_first": false
+    },
+    "model_1_multimer_v2": {
+      "data.transform.common.max_extra_msa": {
+        "eval": 1152,
+        "finetune": 1152,
+        "predict": 1152,
+        "train": 1152
+      },
+      "data.transform.common.max_msa_clusters": {
+        "eval": 128,
+        "finetune": 512,
+        "predict": 252,
+        "train": 128
+      },
+      "globals.is_multimer": true,
+      "globals.use_templates": true,
+      "model.auxiliary_heads.masked_msa_head_c_out": 22,
+      "model.auxiliary_heads.tm_enabled": true,
+      "model.evoformer_stack.opm_first": true,
+      "model.extra_msa_stack.opm_first": true,
+      "model.feature_embedder.c_taf": 34,
+      "model.feature_embedder.tf_dim": 21,
+      "model.loss.supervised_chi_loss.weight": 3.0,
+      "model.structure_module.ipa_scalar_bias": false,
+      "model.structure_module.position_scale": 20,
+      "model.structure_module.separate_kv": true,
+      "model.template_pair_stack.attention_first": false
+    },
+    "model_1_multimer_v3": {
+      "data.transform.common.max_extra_msa": {
+        "eval": 1152,
+        "finetune": 1152,
+        "predict": 2048,
+        "train": 1152
+      },
+      "data.transform.common.max_msa_clusters": {
+        "eval": 128,
+        "finetune": 512,
+        "predict": 508,
+        "train": 128
+      },
+      "globals.is_multimer": true,
+      "globals.use_templates": true,
+      "model.auxiliary_heads.masked_msa_head_c_out": 22,
+      "model.auxiliary_heads.tm_enabled": true,
+      "model.evoformer_stack.fused": true,
+      "model.evoformer_stack.opm_first": true,
+      "model.extra_msa_stack.fused": true,
+      "model.extra_msa_stack.opm_first": true,
+      "model.feature_embedder.c_taf": 34,
+      "model.feature_embedder.tf_dim": 21,
+      "model.structure_module.ipa_scalar_bias": false,
+      "model.structure_module.position_scale": 20,
+      "model.structure_module.separate_kv": true,
+      "model.template_pair_stack.attention_first": false
+    },
+    "model_1_ptm": {
+      "data.transform.common.max_extra_msa.predict": 5120,
+      "data.transform.common.max_extra_msa.train": 5120,
+      "data.transform.use_templates": true,
+      "globals.use_templates": true,
+      "model.auxiliary_heads.tm_enabled": true
+    },
+    "model_2": {
+      "globals.use_templates": true
+    },
+    "model_2_multimer": {
+      "data.transform.common.max_extra_msa": {
+        "eval": 1152,
+        "finetune": 1152,
+        "predict": 1152,
+        "train": 1152
+      },
+      "data.transform.common.max_msa_clusters": {
+        "eval": 128,
+        "finetune": 252,
+        "predict": 128,
+        "train": 128
+      },
+      "globals.is_multimer": true,
+      "globals.use_templates": true,
+      "model.auxiliary_heads.masked_msa_head_c_out": 22,
+      "model.auxiliary_heads.tm_enabled": true,
+      "model.evoformer_stack.opm_first": true,
+      "model.extra_msa_stack.opm_first": true,
+      "model.feature_embedder.c_taf": 34,
+      "model.feature_embedder.tf_dim": 21,
+      "model.loss.supervised_chi_loss.weight": 3.0,
+      "model.structure_module.ipa_scalar_bias": false,
+      "model.structure_module.position_scale": 20,
+      "model.structure_module.separate_kv": true,
+      "model.template_pair_stack.attention_first": false
+    },
+    "model_2_multimer_v2": {
+      "data.transform.common.max_extra_msa": {
+        "eval": 1152,
+        "finetune": 1152,
+        "predict": 1152,
+        "train": 1152
+      },
+      "data.transform.common.max_msa_clusters": {
+        "eval": 128,
+        "finetune": 512,
+        "predict": 252,
+        "train": 128
+      },
+      "globals.is_multimer": true,
+      "globals.use_templates": true,
+      "model.auxiliary_heads.masked_msa_head_c_out": 22,
+      "model.auxiliary_heads.tm_enabled": true,
+      "model.evoformer_stack.opm_first": true,
+      "model.extra_msa_stack.opm_first": true,
+      "model.feature_embedder.c_taf": 34,
+      "model.feature_embedder.tf_dim": 21,
+      "model.loss.supervised_chi_loss.weight": 3.0,
+      "model.structure_module.ipa_scalar_bias": false,
+      "model.structure_module.position_scale": 20,
+      "model.structure_module.separate_kv": true,
+      "model.template_pair_stack.attention_first": false
+    },
+    "model_2_multimer_v3": {
+      "data.transform.common.max_extra_msa": {
+        "eval": 1152,
+        "finetune": 1152,
+        "predict": 2048,
+        "train": 1152
+      },
+      "data.transform.common.max_msa_clusters": {
+        "eval": 128,
+        "finetune": 512,
+        "predict": 508,
+        "train": 128
+      },
+      "globals.is_multimer": true,
+      "globals.use_templates": true,
+      "model.auxiliary_heads.masked_msa_head_c_out": 22,
+      "model.auxiliary_heads.tm_enabled": true,
+      "model.evoformer_stack.fused": true,
+      "model.evoformer_stack.opm_first": true,
+      "model.extra_msa_stack.fused": true,
+      "model.extra_msa_stack.opm_first": true,
+      "model.feature_embedder.c_taf": 34,
+      "model.feature_embedder.tf_dim": 21,
+      "model.structure_module.ipa_scalar_bias": false,
+      "model.structure_module.position_scale": 20,
+      "model.structure_module.separate_kv": true,
+      "model.template_pair_stack.attention_first": false
+    },
+    "model_2_ptm": {
+      "globals.use_templates": true,
+      "model.auxiliary_heads.tm_enabled": true
+    },
+    "model_3": {
+      "data.transform.common.max_extra_msa.predict": 5120,
+      "data.transform.common.max_extra_msa.train": 5120
+    },
+    "model_3_multimer": {
+      "data.transform.common.max_extra_msa": {
+        "eval": 1152,
+        "finetune": 1152,
+        "predict": 1152,
+        "train": 1152
+      },
+      "data.transform.common.max_msa_clusters": {
+        "eval": 128,
+        "finetune": 252,
+        "predict": 128,
+        "train": 128
+      },
+      "globals.is_multimer": true,
+      "globals.use_templates": true,
+      "model.auxiliary_heads.masked_msa_head_c_out": 22,
+      "model.auxiliary_heads.tm_enabled": true,
+      "model.evoformer_stack.opm_first": true,
+      "model.extra_msa_stack.opm_first": true,
+      "model.feature_embedder.c_taf": 34,
+      "model.feature_embedder.tf_dim": 21,
+      "model.loss.supervised_chi_loss.weight": 3.0,
+      "model.structure_module.ipa_scalar_bias": false,
+      "model.structure_module.position_scale": 20,
+      "model.structure_module.separate_kv": true,
+      "model.template_pair_stack.attention_first": false
+    },
+    "model_3_multimer_v2": {
+      "data.transform.common.max_extra_msa": {
+        "eval": 1152,
+        "finetune": 1152,
+        "predict": 1152,
+        "train": 1152
+      },
+      "data.transform.common.max_msa_clusters": {
+        "eval": 128,
+        "finetune": 512,
+        "predict": 252,
+        "train": 128
+      },
+      "globals.is_multimer": true,
+      "globals.use_templates": true,
+      "model.auxiliary_heads.masked_msa_head_c_out": 22,
+      "model.auxiliary_heads.tm_enabled": true,
+      "model.evoformer_stack.opm_first": true,
+      "model.extra_msa_stack.opm_first": true,
+      "model.feature_embedder.c_taf": 34,
+      "model.feature_embedder.tf_dim": 21,
+      "model.loss.supervised_chi_loss.weight": 3.0,
+      "model.structure_module.ipa_scalar_bias": false,
+      "model.structure_module.position_scale": 20,
+      "model.structure_module.separate_kv": true,
+      "model.template_pair_stack.attention_first": false
+    },
+    "model_3_multimer_v3": {
+      "data.transform.common.max_extra_msa": {
+        "eval": 1152,
+        "finetune": 1152,
+        "predict": 2048,
+        "train": 1152
+      },
+      "data.transform.common.max_msa_clusters": {
+        "eval": 128,
+        "finetune": 512,
+        "predict": 508,
+        "train": 128
+      },
+      "globals.is_multimer": true,
+      "globals.use_templates": true,
+      "model.auxiliary_heads.masked_msa_head_c_out": 22,
+      "model.auxiliary_heads.tm_enabled": true,
+      "model.evoformer_stack.fused": true,
+      "model.evoformer_stack.opm_first": true,
+      "model.extra_msa_stack.fused": true,
+      "model.extra_msa_stack.opm_first": true,
+      "model.feature_embedder.c_taf": 34,
+      "model.feature_embedder.tf_dim": 21,
+      "model.structure_module.ipa_scalar_bias": false,
+      "model.structure_module.position_scale": 20,
+      "model.structure_module.separate_kv": true,
+      "model.template_pair_stack.attention_first": false
+    },
+    "model_3_ptm": {
+      "data.transform.common.max_extra_msa.predict": 5120,
+      "data.transform.common.max_extra_msa.train": 5120,
+      "model.auxiliary_heads.tm_enabled": true
+    },
+    "model_4": {
+      "data.transform.common.max_extra_msa.predict": 5120,
+      "data.transform.common.max_extra_msa.train": 5120
+    },
+    "model_4_multimer": {
+      "data.transform.common.max_extra_msa": {
+        "eval": 1152,
+        "finetune": 1152,
+        "predict": 1152,
+        "train": 1152
+      },
+      "data.transform.common.max_msa_clusters": {
+        "eval": 128,
+        "finetune": 252,
+        "predict": 128,
+        "train": 128
+      },
+      "globals.is_multimer": true,
+      "globals.use_templates": true,
+      "model.auxiliary_heads.masked_msa_head_c_out": 22,
+      "model.auxiliary_heads.tm_enabled": true,
+      "model.evoformer_stack.opm_first": true,
+      "model.extra_msa_stack.opm_first": true,
+      "model.feature_embedder.c_taf": 34,
+      "model.feature_embedder.tf_dim": 21,
+      "model.loss.supervised_chi_loss.weight": 3.0,
+      "model.structure_module.ipa_scalar_bias": false,
+      "model.structure_module.position_scale": 20,
+      "model.structure_module.separate_kv": true,
+      "model.template_pair_stack.attention_first": false
+    },
+    "model_4_multimer_v2": {
+      "data.transform.common.max_extra_msa": {
+        "eval": 1152,
+        "finetune": 1152,
+        "predict": 1152,
+        "train": 1152
+      },
+      "data.transform.common.max_msa_clusters": {
+        "eval": 128,
+        "finetune": 512,
+        "predict": 252,
+        "train": 128
+      },
+      "globals.is_multimer": true,
+      "globals.use_templates": true,
+      "model.auxiliary_heads.masked_msa_head_c_out": 22,
+      "model.auxiliary_heads.tm_enabled": true,
+      "model.evoformer_stack.opm_first": true,
+      "model.extra_msa_stack.opm_first": true,
+      "model.feature_embedder.c_taf": 34,
+      "model.feature_embedder.tf_dim": 21,
+      "model.loss.supervised_chi_loss.weight": 3.0,
+      "model.structure_module.ipa_scalar_bias": false,
+      "model.structure_module.position_scale": 20,
+      "model.structure_module.separate_kv": true,
+      "model.template_pair_stack.attention_first": false
+    },
+    "model_4_multimer_v3": {
+      "data.transform.common.max_extra_msa": {
+        "eval": 1152,
+        "finetune": 1152,
+        "predict": 1152,
+        "train": 1152
+      },
+      "data.transform.common.max_msa_clusters": {
+        "eval": 128,
+        "finetune": 512,
+        "predict": 508,
+        "train": 128
+      },
+      "globals.is_multimer": true,
+      "globals.use_templates": true,
+      "model.auxiliary_heads.masked_msa_head_c_out": 22,
+      "model.auxiliary_heads.tm_enabled": true,
+      "model.evoformer_stack.fused": true,
+      "model.evoformer_stack.opm_first": true,
+      "model.extra_msa_stack.fused": true,
+      "model.extra_msa_stack.opm_first": true,
+      "model.feature_embedder.c_taf": 34,
+      "model.feature_embedder.tf_dim": 21,
+      "model.structure_module.ipa_scalar_bias": false,
+      "model.structure_module.position_scale": 20,
+      "model.structure_module.separate_kv": true,
+      "model.template_pair_stack.attention_first": false
+    },
+    "model_4_ptm": {
+      "data.transform.common.max_extra_msa.predict": 5120,
+      "data.transform.common.max_extra_msa.train": 5120,
+      "model.auxiliary_heads.tm_enabled": true
+    },
+    "model_5": {},
+    "model_5_multimer": {
+      "data.transform.common.max_extra_msa": {
+        "eval": 1152,
+        "finetune": 1152,
+        "predict": 1152,
+        "train": 1152
+      },
+      "data.transform.common.max_msa_clusters": {
+        "eval": 128,
+        "finetune": 252,
+        "predict": 128,
+        "train": 128
+      },
+      "globals.is_multimer": true,
+      "globals.use_templates": true,
+      "model.auxiliary_heads.masked_msa_head_c_out": 22,
+      "model.auxiliary_heads.tm_enabled": true,
+      "model.evoformer_stack.opm_first": true,
+      "model.extra_msa_stack.opm_first": true,
+      "model.feature_embedder.c_taf": 34,
+      "model.feature_embedder.tf_dim": 21,
+      "model.loss.supervised_chi_loss.weight": 3.0,
+      "model.structure_module.ipa_scalar_bias": false,
+      "model.structure_module.position_scale": 20,
+      "model.structure_module.separate_kv": true,
+      "model.template_pair_stack.attention_first": false
+    },
+    "model_5_multimer_v2": {
+      "data.transform.common.max_extra_msa": {
+        "eval": 1152,
+        "finetune": 1152,
+        "predict": 1152,
+        "train": 1152
+      },
+      "data.transform.common.max_msa_clusters": {
+        "eval": 128,
+        "finetune": 512,
+        "predict": 252,
+        "train": 128
+      },
+      "globals.is_multimer": true,
+      "globals.use_templates": true,
+      "model.auxiliary_heads.masked_msa_head_c_out": 22,
+      "model.auxiliary_heads.tm_enabled": true,
+      "model.evoformer_stack.opm_first": true,
+      "model.extra_msa_stack.opm_first": true,
+      "model.feature_embedder.c_taf": 34,
+      "model.feature_embedder.tf_dim": 21,
+      "model.loss.supervised_chi_loss.weight": 3.0,
+      "model.structure_module.ipa_scalar_bias": false,
+      "model.structure_module.position_scale": 20,
+      "model.structure_module.separate_kv": true,
+      "model.template_pair_stack.attention_first": false
+    },
+    "model_5_multimer_v3": {
+      "data.transform.common.max_extra_msa": {
+        "eval": 1152,
+        "finetune": 1152,
+        "predict": 1152,
+        "train": 1152
+      },
+      "data.transform.common.max_msa_clusters": {
+        "eval": 128,
+        "finetune": 512,
+        "predict": 508,
+        "train": 128
+      },
+      "globals.is_multimer": true,
+      "globals.use_templates": true,
+      "model.auxiliary_heads.masked_msa_head_c_out": 22,
+      "model.auxiliary_heads.tm_enabled": true,
+      "model.evoformer_stack.fused": true,
+      "model.evoformer_stack.opm_first": true,
+      "model.extra_msa_stack.fused": true,
+      "model.extra_msa_stack.opm_first": true,
+      "model.feature_embedder.c_taf": 34,
+      "model.feature_embedder.tf_dim": 21,
+      "model.structure_module.ipa_scalar_bias": false,
+      "model.structure_module.position_scale": 20,
+      "model.structure_module.separate_kv": true,
+      "model.template_pair_stack.attention_first": false
+    },
+    "model_5_ptm": {
+      "model.auxiliary_heads.tm_enabled": true
+    }
+  },
+  "_config": null,
+  "architectures": [
+    "STFold"
+  ],
+  "mode": "predict",
+  "model_type": "STFold",
+  "stfold_type": "model_5",
+  "torch_dtype": "float32",
+  "transformers_version": "4.31.0.dev0"
+}

model_5/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a99d149c58e32acea75595a6ae951f36c9244ae2d5011a61c6477d0b6ef27327
+size 373385580