Loading version 3 of the ids.txt (Colab)

Open luismrguimaraes opened this issue 1 year ago • 1 comments

Hi, how can we load the version 3 (v3) of the ids.txt files of the Jukebox model in the Colab notebook?

Feb 02 '24 17:02 luismrguimaraes

Apparently the version 3 is used for the model 1b_lyrics, so I changed the variable model to "1b_lyrics", but now I'm having an assertion error when comparing the x.shape with x_cond.shape:

AssertionError: torch.Size([1, 6144, 2048]) != (1, 8192, 2048) nor (1, 1, 2048). Did you pass the correct --sample_length?

Apr 12 '24 16:04 luismrguimaraes