update colab notebook to use log2_mid_count properly

This commit is contained in:
Brett Kuprel 2022-07-04 17:29:44 -04:00
parent 1702d3c439
commit 25f5d8f800

4
min_dalle.ipynb vendored
View File

@ -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",