From 595d15455a0e4b26dc50456e23f9430f28d4382d Mon Sep 17 00:00:00 2001 From: Arthur Holstvoogd Date: Thu, 6 Oct 2022 15:49:35 +0200 Subject: [PATCH] Fix generation of image with s>1000 --- ldm/modules/diffusionmodules/util.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ldm/modules/diffusionmodules/util.py b/ldm/modules/diffusionmodules/util.py index 60b4d8a028..fb671bfb63 100644 --- a/ldm/modules/diffusionmodules/util.py +++ b/ldm/modules/diffusionmodules/util.py @@ -64,6 +64,8 @@ def make_ddim_timesteps( ): if ddim_discr_method == 'uniform': c = num_ddpm_timesteps // num_ddim_timesteps + if c < 1: + c = 1 ddim_timesteps = np.asarray(list(range(0, num_ddpm_timesteps, c))) elif ddim_discr_method == 'quad': ddim_timesteps = (