pin jax version after other installs in mt3 colab; for some reason it…

…'s getting upgraded to 0.4.2 PiperOrigin-RevId: 506951196
yportne8 · Feb 3, 2023 · 66f9619 · 66f9619
1 parent 1ff00b3
commit 66f9619
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/mt3/colab/music_transcription_with_transformers.ipynb b/mt3/colab/music_transcription_with_transformers.ipynb
@@ -65,8 +65,6 @@
  "!apt-get update -qq \u0026\u0026 apt-get install -qq libfluidsynth2 build-essential libasound2-dev libjack-dev\n",
  "\n",
  "!pip install nest-asyncio\n",
- "# pin jax to maintain compatibility with default colab jaxlib version\n",
- "!pip install jax==0.4.1\n",
  "# pin pyfluidsynth to avoid weird octave issue\n",
  "!pip install pyfluidsynth==1.3.0\n",
  "\n",
@@ -80,6 +78,9 @@
  "!mv mt3 mt3_tmp; mv mt3_tmp/* .; rm -r mt3_tmp\n",
  "!python3 -m pip install -e .\n",
  "\n",
+ "# pin jax to maintain compatibility with default colab jaxlib version\n",
+ "!pip install jax==0.4.1\n",
+ "\n",
  "# copy checkpoints\n",
  "!gsutil -q -m cp -r gs:https://mt3/checkpoints .\n",
  "\n",