r
/
stable-diffusion-finetune
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

accept arbitrary s

This commit is contained in:
Robin Rombach 2022-04-04 22:44:51 +02:00 committed by GitHub
parent 32c525011b
commit 5820f690bd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/txt2img.py
View File

@ -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])