diff --git a/ldsrlib/ldm/models/diffusion/ddpm.py b/ldsrlib/ldm/models/diffusion/ddpm.py
index b50c7e0..005178b 100644
--- a/ldsrlib/ldm/models/diffusion/ddpm.py
+++ b/ldsrlib/ldm/models/diffusion/ddpm.py
@@ -16,7 +16,7 @@ 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.rank_zero import rank_zero_only
 
 from ...util import log_txt_as_img, exists, default, ismap, isimage, mean_flat, count_params, instantiate_from_config
 from ...modules.ema import LitEma
diff --git a/requirements.txt b/requirements.txt
index 95e84bd..9bd90a1 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,5 +3,5 @@ torch
 einops
 omegaconf
 tqdm
-pytorch-lightning==1.7.7
+pytorch-lightning
 torchmetrics==0.11.4
\ No newline at end of file