update notebook

2022-07-07 10:28:07 -04:00
parent 27642edbe6
commit 9a4a839b3b
2 changed files with 7 additions and 2 deletions
@@ -130,9 +130,14 @@
      "source": [
        "from IPython.display import display, update_display\n",
        "from math import log2\n",
+        "import torch\n",
        "from min_dalle import MinDalle\n",
        "\n",
-        "model = MinDalle(is_mega=True, is_reusable=True)"
+        "model = MinDalle(\n",
+        "    dtype=torch.float32,\n",
+        "    is_mega=True, \n",
+        "    is_reusable=True\n",
+        ")"
      ]
    },
    {