feat(demo): use vaild model

tools/inference/inference_pipeline.ipynb

@@ -75,7 +75,7 @@
     "# Model references\n",
     "\n",
     "# dalle-mini\n",
-    "DALLE_MODEL = \"dalle-mini/dalle-mini/model-wzoooa1c:latest\"  # can be wandb artifact or 🤗 Hub or local folder or google bucket\n",
+    "DALLE_MODEL = \"dalle-mini/dalle-mini/wzoooa1c:latest\"  # can be wandb artifact or 🤗 Hub or local folder or google bucket\n",
     "DALLE_COMMIT_ID = None\n",
     "\n",
     "# VQGAN model\n",
@@ -272,7 +272,7 @@
    },
    "outputs": [],
    "source": [
-    "prompt = \"emoji of a baby penguin with blue hat, red gloves and green pants\""
+    "prompt = \"sunset over a lake in the mountains\""
    ]
   },
   {
@@ -350,7 +350,7 @@
     "\n",
     "# generate images\n",
     "images = []\n",
-    "for i in trange(min(n_predictions // jax.device_count(), 1)):\n",
+    "for i in trange(max(n_predictions // jax.device_count(), 1)):\n",
     "    # get a new key\n",
     "    key, subkey = jax.random.split(key)\n",
     "    # generate images\n",