Added config presets for esm1b model inference

openfold/config.py

@@ -172,7 +172,7 @@ def model_config(
         c.data.train.crop_size = 384
         c.loss.violation.weight = 1.
         c.loss.experimentally_resolved.weight = 0.01
-    elif name == "seqemb_inference":
+    elif name == "seq_model_esm1b":
         c.data.seqemb_mode.enabled = True
         c.globals.seqemb_mode_enabled = True
         c.model.extra_msa.enabled = False
@@ -181,6 +181,17 @@ def model_config(
         c.data.common.use_template_torsion_angles = True
         c.model.template.enabled = True
         c.data.predict.max_msa_clusters = 1
+    elif name == "seq_model_esm1b_ptm":
+        c.data.seqemb_mode.enabled = True
+        c.globals.seqemb_mode_enabled = True
+        c.model.extra_msa.enabled = False
+        c.model.evoformer_stack.no_column_attention = True
+        c.data.common.use_templates = True
+        c.data.common.use_template_torsion_angles = True
+        c.model.template.enabled = True
+        c.data.predict.max_msa_clusters = 1
+        c.model.heads.tm.enabled = True
+        c.loss.tm.weight = 0.1
     else:
         raise ValueError("Invalid model name")