diff --git a/colab-notebooks/Multitrack_MusicVAE.ipynb b/colab-notebooks/Multitrack_MusicVAE.ipynb index 6408b3b..761bfe6 100644 --- a/colab-notebooks/Multitrack_MusicVAE.ipynb +++ b/colab-notebooks/Multitrack_MusicVAE.ipynb @@ -53,9 +53,8 @@ "source": [ "#@title Setup Environment\n", "\n", - "print('Copying checkpoints and modified SGM SoundFont (https://sites.google.com/site/soundfonts4u) from GCS.')\n", + "print('Copying modified SGM SoundFont (https://sites.google.com/site/soundfonts4u) from GCS.')\n", "print('This will take a few minutes...')\n", - "!gsutil -q -m cp gs://download.magenta.tensorflow.org/models/music_vae/multitrack/* /content/\n", "!gsutil -q -m cp gs://download.magenta.tensorflow.org/soundfonts/SGM-v2.01-Sal-Guit-Bass-V1.3.sf2 /content/\n", "\n", "print('Installing dependencies...')\n", @@ -184,7 +183,7 @@ "config = configs.CONFIG_MAP['hier-multiperf_vel_1bar_med_chords']\n", "model = TrainedModel(\n", " config, batch_size=BATCH_SIZE,\n", - " checkpoint_dir_or_path='/content/model_chords_fb64.ckpt')" + " checkpoint_dir_or_path='gs://download.magenta.tensorflow.org/models/music_vae/multitrack/model_chords_fb64.ckpt')" ], "execution_count": 0, "outputs": [] @@ -335,7 +334,7 @@ "config = configs.CONFIG_MAP['hier-multiperf_vel_1bar_med']\n", "model = TrainedModel(\n", " config, batch_size=BATCH_SIZE,\n", - " checkpoint_dir_or_path='/content/model_fb256.ckpt')\n", + " checkpoint_dir_or_path='gs://download.magenta.tensorflow.org/models/music_vae/multitrack/model_fb256.ckpt')\n", "model._config.data_converter._max_tensors_per_input = None" ], "execution_count": 0,