fixed a typo that introduced a crash

This commit is contained in:
Lincoln Stein 2022-08-18 13:47:07 -04:00
parent 74f238d310
commit 2b2ebd19e7

View File

@ -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,