hanshounsu

cannot import name 'optim' from 'flax' error

2

Hello, I am currently trying the demo, synthesize_midi.ipynb, and when executing the below line ```python from music_spectrogram_diffusion.models.diffusion import models ``` the error from the title occurs: ImportError: cannot import name...

Consistently getting CUDNN_STATUS_INTERNAL_ERROR during second_stage training after diff_epoch

5

I'm currently training the first_stage and second_stage on 2 4090GPUs. Currently the following error is randomly occurring during the second_stage, when epoch > diff_epoch; Traceback (most recent call last): File...