From 592492395a7cafef82588cabb3c25e7b13ea1f8e Mon Sep 17 00:00:00 2001
From: novarobot <88540431+novarobot@users.noreply.github.com>
Date: Tue, 21 Mar 2023 20:28:37 +0100
Subject: [PATCH] Update ddpm.py

Change:
from pytorch_lightning.utilities.distributed import rank_zero_only
to:
from pytorch_lightning.utilities.rank_zero import rank_zero_only
---
 ldm/models/diffusion/ddpm.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ldm/models/diffusion/ddpm.py b/ldm/models/diffusion/ddpm.py
index bbedd04..bd71b1d 100644
--- a/ldm/models/diffusion/ddpm.py
+++ b/ldm/models/diffusion/ddpm.py
@@ -16,8 +16,8 @@ from contextlib import contextmanager
 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.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
 from ldm.modules.distributions.distributions import normal_kl, DiagonalGaussianDistribution