tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

update colab notebook to use log2_mid_count properly

Browse Source
This commit is contained in:
Brett Kuprel 2022-07-04 17:29:44 -04:00
parent 1702d3c439
commit 25f5d8f800
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
min_dalle.ipynb vendored
View File

@ -144,7 +144,7 @@
}, },
"source": [ "source": [
"### Generate Images\n", "### 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", "text = \"Dali painting of WALL·E\" #@param {type:\"string\"}\n",
"grid_size = 3 #@param {type:\"integer\"}\n", "grid_size = 3 #@param {type:\"integer\"}\n",
"seed = -1 #@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", "display_size = 512 #@param {type:\"integer\"}\n",
"\n", "\n",
"image_shape = (display_size, display_size, 3)\n", "image_shape = (display_size, display_size, 3)\n",