From 05b8de53009509ef4e607cc2cabbf5e273a20f88 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Thu, 27 Oct 2022 23:12:21 -0400 Subject: [PATCH] fix --hires to support inpainting model --- ldm/invoke/generator/txt2img2img.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/ldm/invoke/generator/txt2img2img.py b/ldm/invoke/generator/txt2img2img.py index 7762d2a51d..759ba2dba4 100644 --- a/ldm/invoke/generator/txt2img2img.py +++ b/ldm/invoke/generator/txt2img2img.py @@ -25,16 +25,15 @@ class Txt2Img2Img(Generator): kwargs are 'width' and 'height' """ uc, c, extra_conditioning_info = conditioning + scale_dim = min(width, height) + scale = 512 / scale_dim + + init_width = math.ceil(scale * width / 64) * 64 + init_height = math.ceil(scale * height / 64) * 64 @torch.no_grad() def make_image(x_T): - scale_dim = min(width, height) - scale = 512 / scale_dim - - init_width = math.ceil(scale * width / 64) * 64 - init_height = math.ceil(scale * height / 64) * 64 - shape = [ self.latent_channels, init_height // self.downsampling_factor,