About the computation of log probability

[jjd1123]

Hello, I was very impressed by your research. However, I have a question about how you treated the nan value when computing the log probability of $p_{\theta}(x_0|x_1)$. In your code, the variance of this step is zero.