LSTM_Cell dim keeps growing in inference stage

Probably a very silly question but calling the music_inference function multiple time seems to increase the dimensionality of the input LSTM cell. The LSTM cell and other layers have been trained previously and should not change at inference time, but this is not the case. Any hints are welcome.

I experienced the same problem; kept getting error messages when re-running the cells.

My solution so far has been to restart the kernel and rerun all cells above (after implementing my changes which warrants the re-run). Hopefully there is a simpler solution since re-running all cells can become a time consuming process…