How to test the trained diffusion model on our noisy input images?? There is a problem in saving the model also