feat(colab): update mini checkpoint

tools/inference/inference_pipeline.ipynb

@@ -75,12 +75,9 @@
         "# Model references\n",
         "\n",
         "# dalle-mini\n",
-        "DALLE_MODEL = \"dalle-mini/dalle-mini/kvwti2c9:latest\"  # can be wandb artifact or 🤗 Hub or local folder or google bucket\n",
+        "DALLE_MODEL = \"dalle-mini/dalle-mini/mini-1:v0\"  # can be wandb artifact or 🤗 Hub or local folder or google bucket\n",
         "DALLE_COMMIT_ID = None\n",
         "\n",
-        "# dalle-mega - comment this line if your hardware runs out of memory to use dalle-mini instead\n",
-        "DALLE_MODEL = \"dalle-mini/dalle-mini/mega-1-fp16:latest\"\n",
-        "\n",
         "# VQGAN model\n",
         "VQGAN_REPO = \"dalle-mini/vqgan_imagenet_f16_16384\"\n",
         "VQGAN_COMMIT_ID = \"e93a26e7707683d349bf5d5c41c5b0ef69b677a9\"\n",
@@ -90,6 +87,29 @@
         "CLIP_COMMIT_ID = None"
       ]
     },
+    {
+      "cell_type": "markdown",
+      "source": [
+        "If your hardware can handle it, you can use dalle-mega instead of dalle-mini.\n",
+        "\n",
+        "**Note: on free Colab, you will most likely not be able to load dalle-mega so don't run the below cell to use dalle-mini instead.**"
+      ],
+      "metadata": {
+        "id": "Jy01V3OltG5q"
+      }
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "# dalle-mega\n",
+        "DALLE_MODEL = \"dalle-mini/dalle-mini/mega-1-fp16:latest\""
+      ],
+      "metadata": {
+        "id": "VRRX_g9vtMZG"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
     {
       "cell_type": "code",
       "execution_count": null,