From fd818d210dac754e4a9d8c992d32e0ed6804e15d Mon Sep 17 00:00:00 2001 From: Boris Dayma Date: Mon, 21 Aug 2023 22:49:46 -0500 Subject: [PATCH] fix: jax has no extras --- tools/inference/inference_pipeline.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/inference/inference_pipeline.ipynb b/tools/inference/inference_pipeline.ipynb index 92e0f3d15..298a550ff 100644 --- a/tools/inference/inference_pipeline.ipynb +++ b/tools/inference/inference_pipeline.ipynb @@ -47,7 +47,7 @@ "outputs": [], "source": [ "# Required only for colab environments + GPU\n", - "!pip install \"jax[cuda12_pip]==0.3.25\" jaxlib==0.3.25 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html\n", + "!pip install jax==0.3.25 jaxlib==0.3.25 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html\n", "\n", "# Install required libraries\n", "!pip install -q dalle-mini orbax==0.0.23\n",