From a6c07134f3e480a5dc019f1c053c0c98b492e8d6 Mon Sep 17 00:00:00 2001 From: apolinario Date: Wed, 2 Nov 2022 12:55:18 +0100 Subject: [PATCH 1/2] Fix lightning deprecated call --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index e8e18c18f..915505603 100644 --- a/main.py +++ b/main.py @@ -14,7 +14,7 @@ from pytorch_lightning import seed_everything from pytorch_lightning.trainer import Trainer from pytorch_lightning.callbacks import ModelCheckpoint, Callback, LearningRateMonitor -from pytorch_lightning.utilities.distributed import rank_zero_only +from pytorch_lightning.utilities.rank_zero import rank_zero_only from pytorch_lightning.utilities import rank_zero_info from ldm.data.base import Txt2ImgIterableBaseDataset From 13df9ab20b47979d08ea554ceec4b785219f40dc Mon Sep 17 00:00:00 2001 From: apolinario Date: Wed, 2 Nov 2022 12:56:08 +0100 Subject: [PATCH 2/2] Update ddpm.py --- 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