Update comments

chaidiscovery · Feb 12, 2025 · 0bb67dd · 0bb67dd
1 parent 7056f46
commit 0bb67dd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/chai_lab/chai1.py b/chai_lab/chai1.py
@@ -483,14 +483,15 @@ def run_inference(
     fasta_file: Path,
     *,
     output_dir: Path,
+    # Configuration for ESM, MSA, constraints, and templates
     use_esm_embeddings: bool = True,
     use_msa_server: bool = False,
     msa_server_url: str = "https://api.colabfold.com",
     msa_directory: Path | None = None,
     constraint_path: Path | None = None,
     use_templates_server: bool = False,
     template_hits_path: Path | None = None,
-    # expose some params for easy tweaking
+    # Parameters controlling how we do inference
     recycle_msa_subsample: int = 0,
     num_trunk_recycles: int = 3,
     num_diffn_timesteps: int = 200,