diff options
| author | Leonard Kugis <leonard@kug.is> | 2024-04-07 16:54:51 +0200 |
|---|---|---|
| committer | Leonard Kugis <leonard@kug.is> | 2024-04-07 16:54:51 +0200 |
| commit | 6ca61e1a23542c834196f22ae9404e88bc27a5ce (patch) | |
| tree | c0157b47cda4858b11878b6570407d9e542d9394 /modules/models/diffusion/ddpm_edit.py | |
| parent | bef51aed032c0aaa5cfd80445bc4cf0d85b408b5 (diff) | |
Make it compatible with Python < 3.10
Diffstat (limited to 'modules/models/diffusion/ddpm_edit.py')
| -rw-r--r-- | modules/models/diffusion/ddpm_edit.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/models/diffusion/ddpm_edit.py b/modules/models/diffusion/ddpm_edit.py index 6db340da..52656005 100644 --- a/modules/models/diffusion/ddpm_edit.py +++ b/modules/models/diffusion/ddpm_edit.py @@ -19,7 +19,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 ldm.util import log_txt_as_img, exists, default, ismap, isimage, mean_flat, count_params, instantiate_from_config from ldm.modules.ema import LitEma |