From 21e37be8813d4c9e7b33955908ca26cd5e6e4bd5 Mon Sep 17 00:00:00 2001 From: "O. Inha" <50331907+olaviinha@users.noreply.github.com> Date: Tue, 13 Jun 2023 14:37:27 +0300 Subject: [PATCH] rank_zero_only from correct module Fix `ModuleNotFoundError: No module named 'pytorch_lightning.utilities.distributed'` --- 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