feat: remove warnings

dev/inference/inference_pipeline.ipynb

@@ -1441,27 +1441,15 @@
     {
       "cell_type": "code",
       "metadata": {
-        "id": "hiD9ZlHu_PsA",
-        "colab": {
-          "base_uri": "https://localhost:8080/"
-        },
-        "outputId": "a9cb8c25-0271-4d19-a20b-9c2d42f5dd4b"
+        "id": "hiD9ZlHu_PsA"
       },
       "source": [
         "# set up model and processor\n",
         "clip = FlaxCLIPModel.from_pretrained(\"openai/clip-vit-base-patch32\")\n",
         "processor = CLIPProcessor.from_pretrained(\"openai/clip-vit-base-patch32\")"
       ],
-      "execution_count": 19,
-      "outputs": [
-        {
-          "output_type": "stream",
-          "text": [
-            "ftfy or spacy is not installed using BERT BasicTokenizer instead of ftfy.\n"
-          ],
-          "name": "stderr"
-        }
-      ]
+      "execution_count": null,
+      "outputs": []
     },
     {
       "cell_type": "code",
@@ -1758,30 +1746,15 @@
     {
       "cell_type": "code",
       "metadata": {
-        "colab": {
-          "base_uri": "https://localhost:8080/"
-        },
-        "id": "o1n_MF8DkU25",
-        "outputId": "8624c006-414d-4e47-c9c6-dfc6496eaefd"
+        "id": "o1n_MF8DkU25"
       },
       "source": [
         "# replicate parameters on all devices\n",
         "dalle_params = replicate(model.params)\n",
         "vqgan_params = replicate(vqgan.params)"
       ],
-      "execution_count": 26,
-      "outputs": [
-        {
-          "output_type": "stream",
-          "text": [
-            "/usr/local/lib/python3.7/dist-packages/jax/lib/xla_bridge.py:387: UserWarning: jax.host_count has been renamed to jax.process_count. This alias will eventually be removed; please update your code.\n",
-            "  \"jax.host_count has been renamed to jax.process_count. This alias \"\n",
-            "/usr/local/lib/python3.7/dist-packages/jax/lib/xla_bridge.py:374: UserWarning: jax.host_id has been renamed to jax.process_index. This alias will eventually be removed; please update your code.\n",
-            "  \"jax.host_id has been renamed to jax.process_index. This alias \"\n"
-          ],
-          "name": "stderr"
-        }
-      ]
+      "execution_count": null,
+      "outputs": []
     },
     {
       "cell_type": "code",