From e02431f18d246884e8d4afba110732abf7c7551d Mon Sep 17 00:00:00 2001
From: Kadir Barut <34428205+kadirb4rut@users.noreply.github.com>
Date: Fri, 22 Dec 2023 20:45:17 +0300
Subject: [PATCH] Update ddpm.py

Now, rank_zero_only is in /pytorch_lightning/utilities/rank_zero.py

I did the installations on Colab without conda, just pip and ran this code with the latest versions of all required packages and that's how I solved this error.
---
 ldm/models/diffusion/ddpm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ldm/models/diffusion/ddpm.py b/ldm/models/diffusion/ddpm.py
index bbedd04cf..ef0990e10 100644
--- a/ldm/models/diffusion/ddpm.py
+++ b/ldm/models/diffusion/ddpm.py
@@ -16,7 +16,7 @@
 from functools import partial
 from tqdm import tqdm
 from torchvision.utils import make_grid
-from pytorch_lightning.utilities.distributed import rank_zero_only
+from pytorch_lightning.utilities.rank_zero import rank_zero_only
 
 from ldm.util import log_txt_as_img, exists, default, ismap, isimage, mean_flat, count_params, instantiate_from_config
 from ldm.modules.ema import LitEma