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

default float16 in colab

Browse Source
This commit is contained in:
Brett Kuprel
2022-07-10 13:24:52 -04:00
parent cfb9f60b6e
commit e556296b5b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
min_dalle.ipynb vendored
View File

@@ -128,7 +128,7 @@
} }
], ],
"source": [ "source": [
"dtype = \"float32\" #@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", "from math import log2\n",
"import torch\n", "import torch\n",