I ran into this same issue, it was caused by user error, but the error handling of diffusion could still be fixed. Here's how to reproduce: