Google Colab version for Music Synthesis and Finetune problem

I’m totally blown away by the speed and quality of the music generation tool you shared! Thank you so much!

I have started drafting Google Colab for Music Generation, and it works just fine, you can use it here.

However, the Finetuning process runs into a problem, after this command: python musika_train.py --train_path prepared_for_training --load_path /content/musika/checkpoints/misc --lr 0.00004

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/content/musika/musika_train.py", line 22, in <module>
    models_ls = M.initialize_networks()
  File "/content/musika/models.py", line 752, in initialize_networks
    (critic, gen, enc, dec, enc2, dec2, gen_ema, [opt_dec, opt_disc], switch) = self.get_networks()
  File "/content/musika/models.py", line 731, in get_networks
    ) = self.load(self.args.load_path, load_dec=False)
  File "/content/musika/models.py", line 661, in load
    opt_disc.set_weights(np.load(path + "/opt_disc.npy", allow_pickle=True))
  File "/usr/local/lib/python3.9/site-packages/numpy/lib/npyio.py", line 450, in load
    raise IOError(
OSError: Failed to interpret file '/content/musika/checkpoints/misc/opt_disc.npy' as a pickle

I have tried to edit npyio.py where I have allowed all allow_pickle=False parts, but it still does not help.

Thanks again for the amazing research!

[marcoppasini]

Yes I reached the limit super fast, should work now! I will have to upload the optimizer files somewhere else

[marcoppasini]

Now model weights are downloaded from the huggingface Hub, so no more git lfs problems Thank you for reaching out and for the recommendation!