change pinned jax version in transcription colab to match jaxlib

PiperOrigin-RevId: 473354394
yportne8 · Sep 9, 2022 · 571c4d0 · 571c4d0
1 parent fc8ce8c
commit 571c4d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mt3/colab/music_transcription_with_transformers.ipynb b/mt3/colab/music_transcription_with_transformers.ipynb
@@ -87,7 +87,7 @@
  "\n",
  "# downgrade jax\n",
  "# TODO(iansimon): remove this once we can use the latest t5x\n",
- "!pip install jax==0.3.14\n",
+ "!pip install jax==0.3.15\n",
  "\n",
  "# copy checkpoints\n",
  "!gsutil -q -m cp -r gs:https://mt3/checkpoints .\n",