Merge branch 'hillst/pydantic-ux' of github.com:NVIDIA/bionemo-framew…

…ork into hillst/pydantic-ux
NVIDIA · Nov 5, 2024 · bfcb2c9 · bfcb2c9
2 parents 13d56c2 + 0566901
commit bfcb2c9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -338,7 +338,7 @@ the default pretraining DataConfig and DataModule will be insufficient. See ESM2
 > ⚠️ **Warning:** This setup does NO configuration of Weights and Biases. Edit your config JSON and populate it with your WandB details.
 
 ```bash
-bionemo-esm2-train \
+bionemo-geneformer-train \
 --data-config-t bionemo.geneformer.run.config_models.GeneformerPretrainingDataConfig \
 --model-config-t bionemo.geneformer.run.config_models.ExposedGeneformerPretrainConfig \
 --config my_config.json

diff --git a/sub-packages/bionemo-esm2/src/bionemo/esm2/run/main.py b/sub-packages/bionemo-esm2/src/bionemo/esm2/run/main.py
@@ -25,13 +25,13 @@
 
 def main():  # noqa: D103
     def parse_args():
-        parser = argparse.ArgumentParser(description="Run Geneformer pretraining")
+        parser = argparse.ArgumentParser(description="Run ESM2 pretraining")
         parser.add_argument("--config", type=str, required=True, help="Path to the JSON configuration file")
         parser.add_argument(
             "--model-config-t",
             default=ExposedESM2PretrainConfig,
             required=False,
-            help="fully resolvable python import path to the ModelConfig object. Builtin options are ExposedGeneformerPretrainConfig and ExposedFineTuneSeqLenBioBertConfig.",
+            help="fully resolvable python import path to the ModelConfig object. Builtin options are ExposedESM2PretrainConfig.",
         )
         parser.add_argument(
             "--data-config-t",