From 67b6325fbba74040bb7ad6c31d9d86900994f7f2 Mon Sep 17 00:00:00 2001 From: kamil Date: Sat, 22 Jul 2023 12:38:50 +0200 Subject: [PATCH] change utilities.distributed to utilities.rank_zero --- cldm/logger.py | 2 +- ldm/models/diffusion/ddpm.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/cldm/logger.py b/cldm/logger.py index 6a8803846f..b314228763 100644 --- a/cldm/logger.py +++ b/cldm/logger.py @@ -5,7 +5,7 @@ import torchvision from PIL import Image from pytorch_lightning.callbacks import Callback -from pytorch_lightning.utilities.distributed import rank_zero_only +from pytorch_lightning.utilities.rank_zero import rank_zero_only class ImageLogger(Callback): diff --git a/ldm/models/diffusion/ddpm.py b/ldm/models/diffusion/ddpm.py index f71a44af48..39e58f0d37 100644 --- a/ldm/models/diffusion/ddpm.py +++ b/ldm/models/diffusion/ddpm.py @@ -17,7 +17,7 @@ import itertools 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 omegaconf import ListConfig from ldm.util import log_txt_as_img, exists, default, ismap, isimage, mean_flat, count_params, instantiate_from_config