Diffusion_pytorch_modelHD