From fff19bf82ec55946cde7a28374d6cfa5773e3e72 Mon Sep 17 00:00:00 2001 From: rromb Date: Wed, 1 Jun 2022 09:52:17 +0200 Subject: [PATCH] handle listconfig --- ldm/models/diffusion/ddpm.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/ldm/models/diffusion/ddpm.py b/ldm/models/diffusion/ddpm.py index 4881913..06a04bd 100644 --- a/ldm/models/diffusion/ddpm.py +++ b/ldm/models/diffusion/ddpm.py @@ -17,6 +17,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 omegaconf import ListConfig 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 @@ -1188,6 +1189,8 @@ class LatentDiffusion(DDPM): def get_unconditional_conditioning(self, batch_size, null_label=None): if null_label is not None: xc = null_label + if isinstance(xc, ListConfig): + xc = list(xc) if isinstance(xc, dict) or isinstance(xc, list): c = self.get_learned_conditioning(xc) else: