From 21c09e088b9682fca55f6b6340bea9725c6760b9 Mon Sep 17 00:00:00 2001 From: fzhang Date: Sat, 5 Nov 2022 17:23:44 +0800 Subject: [PATCH] fix-outpainting --- ldm/models/diffusion/ddpm.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ldm/models/diffusion/ddpm.py b/ldm/models/diffusion/ddpm.py index bbedd04cf..59d347732 100644 --- a/ldm/models/diffusion/ddpm.py +++ b/ldm/models/diffusion/ddpm.py @@ -1334,14 +1334,16 @@ def log_images(self, batch, N=8, n_row=4, sample=True, ddim_steps=200, ddim_eta= ddim_steps=ddim_steps, x0=z[:N], mask=mask) x_samples = self.decode_first_stage(samples.to(self.device)) log["samples_inpainting"] = x_samples - log["mask"] = mask + log["mask_inpainting"] = mask # outpaint + mask = 1 - mask with self.ema_scope("Plotting Outpaint"): samples, _ = self.sample_log(cond=c, batch_size=N, ddim=use_ddim,eta=ddim_eta, ddim_steps=ddim_steps, x0=z[:N], mask=mask) x_samples = self.decode_first_stage(samples.to(self.device)) log["samples_outpainting"] = x_samples + log["mask_outpainting"] = mask if plot_progressive_rows: with self.ema_scope("Plotting Progressives"):