From cb0414d89bd72416e9bc2a9baa8f11262fa10157 Mon Sep 17 00:00:00 2001 From: Patrick Esser Date: Wed, 1 Jun 2022 09:14:09 +0000 Subject: [PATCH] goooo --- .../stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml | 3 ++- ldm/models/diffusion/ddpm.py | 4 +++- 2 files changed, 5 insertions(+), 2 deletions(-) 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()