Update ddpm.py

Fixed rank_zero_only reference that moved AUTOMATIC1111/stable-diffusion-webui#11458
kjerk · Mar 31, 2024 · fb40480 · fb40480
1 parent cf1d67a
commit fb40480
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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