code running issue, dimension mismatch

[liuzhaoda]

Hello,
I have read your paper published in NeuroIPS in 2021, and thank you very much for publishing the code.
However, during the process of running the code, only main.py was run, and there was a dimension mismatch issue in calculating the elbo. May I ask if you have the updated code in the future.
Kind greetings,
Liu