From e556296b5bf8d7def356c97347cc4bcf2470fb33 Mon Sep 17 00:00:00 2001 From: Brett Kuprel Date: Sun, 10 Jul 2022 13:24:52 -0400 Subject: [PATCH] default float16 in colab --- min_dalle.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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",