https://github.com/NUS-HPC-AI-Lab/Neural-Network-Parameter-Diffusion/blob/d108df0a09a8065479f9e4cca957c0770c83f13b/core/system/ddpm.py#L287 pred_x_0 should be pred_x0.
