diff --git a/configs/stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml b/configs/stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml index ca88ebd..1f29a6d 100644 --- a/configs/stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml +++ b/configs/stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml @@ -110,11 +110,12 @@ lightning: image_logger: target: main.ImageLogger params: - batch_frequency: 50 + batch_frequency: 5000 max_images: 4 increase_log_steps: False log_first_step: False log_images_kwargs: + use_ema_scope: False inpaint: False plot_progressive_rows: False plot_diffusion_rows: False diff --git a/ldm/models/diffusion/ddpm.py b/ldm/models/diffusion/ddpm.py index 29620c1..db9e2d3 100644 --- a/ldm/models/diffusion/ddpm.py +++ b/ldm/models/diffusion/ddpm.py @@ -1194,7 +1194,9 @@ class LatentDiffusion(DDPM): if isinstance(xc, dict) or isinstance(xc, list): c = self.get_learned_conditioning(xc) else: - c = self.get_learned_conditioning(xc.to(self.device)) + if hasattr(xc, "to"): + xc = xc.to(self.device) + c = self.get_learned_conditioning(xc) else: # todo: get null label from cond_stage_model raise NotImplementedError()