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

add temperature parameter

Browse Source
This commit is contained in:
Brett Kuprel 2022-07-11 12:30:10 -04:00
parent d64e957731
commit 9ac2ed2dc7
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
min_dalle.ipynb vendored
View File

@ -130,7 +130,6 @@
"source": [ "source": [
"dtype = \"float16\" #@param [\"float32\", \"float16\", \"bfloat16\"]\n", "dtype = \"float16\" #@param [\"float32\", \"float16\", \"bfloat16\"]\n",
"from IPython.display import display, update_display\n", "from IPython.display import display, update_display\n",
"from math import log2\n",
"import torch\n", "import torch\n",
"from min_dalle import MinDalle\n", "from min_dalle import MinDalle\n",
"\n", "\n",
@ -204,8 +203,8 @@
" seed=-1,\n", " seed=-1,\n",
" grid_size=grid_size,\n", " grid_size=grid_size,\n",
" log2_mid_count=log2_mid_count,\n", " log2_mid_count=log2_mid_count,\n",
" log2_k=int(log2(top_k)),\n", " top_k=int(top_k),\n",
" log2_supercondition_factor=log2(supercondition_factor)\n", " supercondition_factor=float(supercondition_factor)\n",
")\n", ")\n",
"\n", "\n",
"is_first = True\n", "is_first = True\n",