feat: remove warning

dev/inference/inference_pipeline.ipynb

@@ -1725,30 +1725,15 @@
     {
       "cell_type": "code",
       "metadata": {
-        "id": "o1n_MF8DkU25",
-        "outputId": "1ddf7fdd-a0f9-4273-bd0a-5ce5f508a2df",
-        "colab": {
-          "base_uri": "https://localhost:8080/"
-        }
+        "id": "o1n_MF8DkU25"
       },
       "source": [
         "# replicate parameters on all devices\n",
         "dalle_params = replicate(model.params)\n",
         "vqgan_params = replicate(vqgan.params)"
       ],
-      "execution_count": 24,
-      "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",