default float16 in colab

2 years ago · e556296b5b
parent cfb9f60b6e
commit e556296b5b
1 changed files with 1 additions and 1 deletions
--- 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",