From 2b2ebd19e70d5b127de4bda460740ad4970fefc3 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Thu, 18 Aug 2022 13:47:07 -0400 Subject: [PATCH] fixed a typo that introduced a crash --- ldm/simplet2i.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ldm/simplet2i.py b/ldm/simplet2i.py index bcd3e5a8e6..adcbd6372a 100644 --- a/ldm/simplet2i.py +++ b/ldm/simplet2i.py @@ -197,7 +197,7 @@ class T2I: shape = [self.latent_channels, height // self.downsampling_factor, width // self.downsampling_factor] samples_ddim, _ = sampler.sample(S=steps, conditioning=c, - batch_size_size=batch_size, + batch_size=batch_size, shape=shape, verbose=False, unconditional_guidance_scale=cfg_scale,