accept arbitrary s
This commit is contained in:
parent
32c525011b
commit
5820f690bd
1 changed files with 1 additions and 1 deletions
|
@ -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])
|
||||
|
|
Loading…
Reference in a new issue