diff --git a/min_dalle.ipynb b/min_dalle.ipynb
index 8819bf2..2162c7f 100644
--- a/min_dalle.ipynb
+++ b/min_dalle.ipynb
@@ -144,7 +144,7 @@
       },
       "source": [
         "### Generate Images\n",
-        "Note: reduce the grid size if you run out of GPU memory.  4x4 has been tested to work on T4 and P100 (with intermediate_image_count = None)"
+        "Note: reduce the grid size if you run out of GPU memory.  4x4 has been tested to work on T4 and P100 (with intermediate_image_count = 1)"
       ]
     },
     {
@@ -184,7 +184,7 @@
         "text = \"Dali painting of WALL·E\" #@param {type:\"string\"}\n",
         "grid_size = 3 #@param {type:\"integer\"}\n",
         "seed = -1 #@param {type:\"integer\"}\n",
-        "intermediate_image_count = 8 #@param [\"2\", \"4\", \"8\", \"16\", \"None\"] {type:\"raw\"}\n",
+        "intermediate_image_count = 8 #@param [\"1\", \"2\", \"4\", \"8\", \"16\"] {type:\"raw\"}\n",
         "display_size = 512 #@param {type:\"integer\"}\n",
         "\n",
         "image_shape = (display_size, display_size, 3)\n",