diff --git a/min_dalle.ipynb b/min_dalle.ipynb
index f8236f9..c029e63 100644
--- a/min_dalle.ipynb
+++ b/min_dalle.ipynb
@@ -128,7 +128,7 @@
         }
       ],
       "source": [
-        "dtype = \"float32\" #@param [\"float32\", \"float16\", \"bfloat16\"]\n",
+        "dtype = \"float16\" #@param [\"float32\", \"float16\", \"bfloat16\"]\n",
         "from IPython.display import display, update_display\n",
         "from math import log2\n",
         "import torch\n",