diff --git a/notebook.ipynb b/notebook.ipynb
index 94e55f2..55f96d2 100644
--- a/notebook.ipynb
+++ b/notebook.ipynb
@@ -55,7 +55,7 @@
         "    # Check what we're starting with (Issue 399)\n",
         "    !if [ ! -d logs ]; then mkdir logs; fi\n",
         "    !pip list > logs/packages.log\n",
-        "    !pip install neural-amp-modeler==0.12 >& logs/install.log\n",
+        "    !pip install neural-amp-modeler==0.13 >& logs/install.log\n",
         "    # Hint: use the next line instead for the very latest!\n",
         "    # !pip install git+https://github.com/sdatkinson/neural-amp-modeler.git@main\n",
         "\n",
@@ -74,7 +74,6 @@
         "\n",
         "#@markdown # Training parameters\n",
         "epochs = 100 #@param {type: \"number\"}\n",
-        "architecture = \"standard\"  #@param [\"standard\", \"lite\", \"feather\", \"nano\"] {type: \"string\"}\n",
         "latency_samples = \"auto\"  #@param {type: \"string\"}\n",
         "ignore_checks = False #@param {type: \"boolean\"}\n",
         "\n",
@@ -144,7 +143,6 @@
         "%tensorboard --logdir /content/lightning_logs\n",
         "run(\n",
         "    epochs=epochs,\n",
-        "    architecture=architecture,\n",
         "    ignore_checks=ignore_checks,\n",
         "    user_metadata=user_metadata,\n",
         "    delay=_parse_latency(latency_samples)\n",
@@ -200,4 +198,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
+}
\ No newline at end of file