From b52098eb4764503d932fdc83c67165b5533eb92d Mon Sep 17 00:00:00 2001 From: Rerange Date: Sun, 14 Aug 2022 19:10:29 +0800 Subject: [PATCH 1/2] fix module 'jax' has no attribute 'tree_multimap' Set the jax package to the correct version. --- mt3/colab/music_transcription_with_transformers.ipynb | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mt3/colab/music_transcription_with_transformers.ipynb b/mt3/colab/music_transcription_with_transformers.ipynb index f885e47..8151852 100644 --- a/mt3/colab/music_transcription_with_transformers.ipynb +++ b/mt3/colab/music_transcription_with_transformers.ipynb @@ -70,6 +70,8 @@ "!pip install clu==0.0.7\n", "# pin Orbax to use Checkpointer\n", "!pip install orbax==0.0.2\n", + "#Install the correct version of the jax package\n", + "!pip install jax==0.3.15\n", "\n", "# install t5x\n", "!git clone --branch=main https://github.com/google-research/t5x\n", From dc09a8e81d60138f38be9c13002208026b1b1b1b Mon Sep 17 00:00:00 2001 From: Rerange Date: Sun, 14 Aug 2022 21:37:28 +0800 Subject: [PATCH 2/2] Update music_transcription_with_transformers.ipynb --- mt3/colab/music_transcription_with_transformers.ipynb | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/mt3/colab/music_transcription_with_transformers.ipynb b/mt3/colab/music_transcription_with_transformers.ipynb index 8151852..f6fa61c 100644 --- a/mt3/colab/music_transcription_with_transformers.ipynb +++ b/mt3/colab/music_transcription_with_transformers.ipynb @@ -70,8 +70,6 @@ "!pip install clu==0.0.7\n", "# pin Orbax to use Checkpointer\n", "!pip install orbax==0.0.2\n", - "#Install the correct version of the jax package\n", - "!pip install jax==0.3.15\n", "\n", "# install t5x\n", "!git clone --branch=main https://github.com/google-research/t5x\n", @@ -92,6 +90,9 @@ "# copy soundfont (originally from https://sites.google.com/site/soundfonts4u)\n", "!gsutil -q -m cp gs://magentadata/soundfonts/SGM-v2.01-Sal-Guit-Bass-V1.3.sf2 .\n", "\n", + "#Install the correct version of the jax package\n", + "!pip install jax==0.3.15\n", + "\n", "import json\n", "import IPython\n", "\n",