Loading version 3 of the ids.txt (Colab)

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

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?