diff --git a/DFL_Colab.ipynb b/DFL_Colab.ipynb
index 6fadb9c..fe4129e 100644
--- a/DFL_Colab.ipynb
+++ b/DFL_Colab.ipynb
@@ -689,7 +689,7 @@
       },
       "source": [
         "#@title Training\n",
-        "Model = \"SAEHD\" #@param [\"SAEHD\", \"Quick96\", \"XSeg\"]\n",
+        "Model = \"SAEHD\" #@param [\"SAEHD\", \"AMP\", \"Quick96\", \"XSeg\"]\n",
         "Backup_every_hour = True #@param {type:\"boolean\"}\n",
         "Silent_Start = True #@param {type:\"boolean\"}\n",
         "\n",