AttributeError: 'GaussianDiffusion' object has no attribute 'module'

[Amar Kumar]

Cold-Diffusion-Models/deblurring-diffusion-pytorch/deblurring_diffusion_pytorch/deblurring_diffusion_pytorch.py

Line 1241 in f8b1379

 X_0s, X_ts = self.ema_model.module.all_sample(batch_size=batches, img=og_img, times=s_times) # change for DP 

Modified this to:
X_0s, X_ts = self.ema_model.all_sample(batch_size=batches, img=og_img, times=s_times) # change for DP

seems to solve the error.

[Arpit Bansal]

Yes, .module is there because of data parallel.