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

simplify colab

Browse Source
This commit is contained in:
Brett Kuprel 2022-07-05 09:37:52 -04:00
parent e1c744821b
commit 7cae9f27be
1 changed files with 14 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
min_dalle.ipynb vendored
View File

@ -3,8 +3,8 @@
{
"cell_type": "markdown",
"metadata": {
"colab_type": "text",
"id": "view-in-github"
"id": "view-in-github",
"colab_type": "text"
},
"source": [
"<a href=\"https://colab.research.google.com/github/kuprel/min-dalle/blob/main/min_dalle.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
@ -30,7 +30,7 @@
},
{
"cell_type": "code",
"execution_count": 1,
"execution_count": null,
"metadata": {
"cellView": "code",
"colab": {
@ -102,7 +102,7 @@
},
{
"cell_type": "code",
"execution_count": 2,
"execution_count": null,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
@ -145,18 +145,16 @@
"source": [
"### Generate Images\n",
"\n",
"- `grid_size` Size of the image grid. 3x3 works best when displaying intermediate results. 4x4 has been tested to work on T4 and P100 with intermediate_image_count = 1\n",
"- `grid_size` Size of the image grid. 3x3 works best when displaying intermediate outputs. 4x4 has been tested to work on T4 and P100 with intermediate outputs off\n",
"\n",
"- `seed` Set the seed to a positive number for reproducible results\n",
"\n",
"- `intermediate_image_count` Set the total number of images to show while running. Each image adds a slight delay.\n",
"- `intermediate_outputs` Whether to show intermediate output. Adds a small delay and increases memory usage.\n",
"\n",
"- `supercondition_factor` Higher values result in better agreement with the text but a narrower variety of generated images"
]
},
{
"cell_type": "code",
"execution_count": 4,
"execution_count": null,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/",
@ -189,16 +187,16 @@
"%%time\n",
"\n",
"text = \"Dali painting of WALL·E\" #@param {type:\"string\"}\n",
"intermediate_outputs = True #@param {type:\"boolean\"}\n",
"grid_size = 3 #@param {type:\"integer\"}\n",
"seed = -1 #@param {type:\"integer\"}\n",
"intermediate_image_count = 4 #@param [\"1\", \"2\", \"4\", \"8\", \"16\"] {type:\"raw\"}\n",
"supercondition_factor = 8 #@param [\"2\", \"4\", \"8\", \"16\", \"32\", \"64\"] {type:\"raw\"}\n",
"log2_mid_count = 3 if intermediate_outputs else 0\n",
"\n",
"image_stream = model.generate_image_stream(\n",
" text=text,\n",
" seed=seed,\n",
" seed=-1,\n",
" grid_size=grid_size,\n",
" log2_mid_count=log2(intermediate_image_count),\n",
" log2_mid_count=log2_mid_count,\n",
" log2_supercondition_factor=log2(supercondition_factor)\n",
")\n",
"\n",
@ -216,9 +214,9 @@
"collapsed_sections": [
"Zl_ZFisFApeh"
],
"include_colab_link": true,
"name": "min-dalle",
"provenance": []
"provenance": [],
"include_colab_link": true
},
"gpuClass": "standard",
"kernelspec": {
@ -231,4 +229,4 @@
},
"nbformat": 4,
"nbformat_minor": 0
}
}