From f9501ecc4c4ba47a8cf1be8535e7c8dca94cc4c7 Mon Sep 17 00:00:00 2001 From: flowtone Date: Thu, 15 Feb 2024 09:36:44 +0200 Subject: [PATCH] upgrade pytorch lightning --- ldsrlib/ldm/models/diffusion/ddpm.py | 2 +- requirements.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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