diff --git a/sub-packages/bionemo-evo2/examples/zeroshot_brca1.ipynb b/sub-packages/bionemo-evo2/examples/zeroshot_brca1.ipynb index f5cd3b309..ebbf21bb7 100644 --- a/sub-packages/bionemo-evo2/examples/zeroshot_brca1.ipynb +++ b/sub-packages/bionemo-evo2/examples/zeroshot_brca1.ipynb @@ -562,16 +562,14 @@ "outputs": [], "source": [ "%%capture\n", - "EXPERIMENTAL_1b_CHECKPOINT: bool = False\n", "MODEL_SIZE = \"1b\" # also try 7b if you have a GPU with more than 32GB of memory\n", + "\n", "# Define checkpoint path\n", - "if EXPERIMENTAL_1b_CHECKPOINT and MODEL_SIZE == \"1b\":\n", + "if MODEL_SIZE == \"1b\":\n", " from bionemo.core.data.load import load\n", "\n", - " # This is a new 1b checkpoint that has been fine-tuned on BF16 hardware. It should be able to handle FP8 as well.\n", - " # this line will download the checkpoint from NGC to your $HOME/.cache/bionemo directory and return the path.\n", - " # alternatively you can use `CHECKPOINT_PATH=$(download_bionemo_data evo2/1b-8k-bf16:1.0)` to do the same on the\n", - " # command line.\n", + " # This line will download the checkpoint from NGC to your $HOME/.cache/bionemo directory and return the path.\n", + " # To do the same from the command line, use `CHECKPOINT_PATH=$(download_bionemo_data evo2/1b-8k-bf16:1.0)`\n", " checkpoint_path = load(\"evo2/1b-8k-bf16:1.0\")\n", "else:\n", " checkpoint_path = Path(f\"nemo2_evo2_{MODEL_SIZE}_8k\")\n", @@ -1186,7 +1184,7 @@ "\n", "| Model Size | Dataset Sample Fraction | AUROC |\n", "|------------|------------------------|-------|\n", - "| Evo 2 1B | 100% | 0.76 |\n", + "| Evo 2 1B | 100% | 0.74 |\n", "| Evo 2 7B | 100% | 0.87 |\n" ] }