Added model sizes to notebooks for clarity

vincentamato · vincentamato · commit 05588e50aee0 · 2025-08-16T12:22:57.000-04:00
diff --git a/esm/notebooks/contact_prediction.ipynb b/esm/notebooks/contact_prediction.ipynb
diff --git a/esm/notebooks/embeddings.ipynb b/esm/notebooks/embeddings.ipynb
@@ -87,7 +87,7 @@
    "source": [
     "### Loading the model and tokenizing a sequence\n",
     "\n",
-    "First, load the ESM-2 model. Change the path below to point to your converted checkpoint.\n",
+    "Load the ESM-2 model.Here we will use the 650M parameter version. Change the path below to point to your converted checkpoint.\n",
     "\n"
    ]
   },
diff --git a/esm/notebooks/mutation_effect_prediction.ipynb b/esm/notebooks/mutation_effect_prediction.ipynb
@@ -82,7 +82,7 @@
    "id": "52c8a805",
    "metadata": {},
    "source": [
-    "Load the ESM-2 model. Change the path below to point to your converted checkpoint."
+    "Load the ESM-2 model. Here we will use the 650M parameter version. Change the path below to point to your converted checkpoint."
    ]
   },
   {

Original file line number	Diff line number	Diff line change
`@@ -87,7 +87,7 @@`
`87`	`87`	`"source": [`
`88`	`88`	`"### Loading the model and tokenizing a sequence\n",`
`89`	`89`	`"\n",`
`90`		`- "First, load the ESM-2 model. Change the path below to point to your converted checkpoint.\n",`
	`90`	`+ "Load the ESM-2 model.Here we will use the 650M parameter version. Change the path below to point to your converted checkpoint.\n",`
`91`	`91`	`"\n"`
`92`	`92`	`]`
`93`	`93`	`},`
Original file line number	Diff line number	Diff line change
`@@ -82,7 +82,7 @@`
`82`	`82`	`"id": "52c8a805",`
`83`	`83`	`"metadata": {},`
`84`	`84`	`"source": [`
`85`		`- "Load the ESM-2 model. Change the path below to point to your converted checkpoint."`
	`85`	`+ "Load the ESM-2 model. Here we will use the 650M parameter version. Change the path below to point to your converted checkpoint."`
`86`	`86`	`]`
`87`	`87`	`},`
`88`	`88`	`{`