From 5820f690bda6ea64616dd3197b8bb5bca4f2eb7d Mon Sep 17 00:00:00 2001 From: Robin Rombach Date: Mon, 4 Apr 2022 22:44:51 +0200 Subject: [PATCH] accept arbitrary s --- scripts/txt2img.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/txt2img.py b/scripts/txt2img.py index fdfc1e6..dc37752 100644 --- a/scripts/txt2img.py +++ b/scripts/txt2img.py @@ -118,7 +118,7 @@ if __name__ == "__main__": with torch.no_grad(): with model.ema_scope(): uc = None - if opt.scale > 0: + if opt.scale != 1.0: uc = model.get_learned_conditioning(opt.n_samples * [""]) for n in trange(opt.n_iter, desc="Sampling"): c = model.get_learned_conditioning(opt.n_samples * [prompt])