pin clu and t5x in colab for python 3.7 compatibility

PiperOrigin-RevId: 454263760
yportne8 · Jun 10, 2022 · 391b87b · 391b87b
1 parent c33cf12
commit 391b87b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mt3/colab/music_transcription_with_transformers.ipynb b/mt3/colab/music_transcription_with_transformers.ipynb
@@ -67,9 +67,13 @@
  "!pip install --upgrade pip\n",
  "!pip install nest-asyncio\n",
  "!pip install pyfluidsynth\n",
+ "# pin CLU for python 3.7 compatibility\n",
+ "!pip install clu==0.0.7\n",
  "\n",
  "# install t5x\n",
  "!git clone --branch=main https://github.com/google-research/t5x\n",
+ "# pin T5X for python 3.7 compatibility\n",
+ "!cd t5x; git reset --hard 2e05ad41778c25521738418de805757bf2e41e9e; cd ..\n",
  "!mv t5x t5x_tmp; mv t5x_tmp/* .; rm -r t5x_tmp\n",
  "!sed -i 's:jax\\[tpu\\]:jax:' setup.py\n",
  "!python3 -m pip install -e .\n",