Matteo Pozzi

Hi @tuttyfrutyee, i presume you are referring to [this](https://github.com/lucidrains/denoising-diffusion-pytorch/blob/main/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py#L1004-L1005) file and [these](https://github.com/lucidrains/denoising-diffusion-pytorch/blob/0dee2d8eebe251e666ef60c1b48d3333a18f4065/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py#L1004-L1005) lines which i also report here: ```python if accelerator.is_main_process: self.ema.update() ``` therefore ema.update calls [this](https://github.com/lucidrains/ema-pytorch/blob/abb558d49839d1aed8ca72e1d5abf7bfe0a39e95/ema_pytorch/ema_pytorch.py#L133) and the following...

Hi, it seems that the same problem also occur in the `train_text_to_image.py` script: I am using a SLURM cluster with 4 K-80 with 48GB of RAM in a single node....

thanks @williamberman and @sayakpaul , An update here: I can confirm that using the LoRA script worked out just fine. I managed to make it run on 2 1080 with...