artem/stable-diffusion
1
0
Fork 0
mirror of https://github.com/CompVis/stable-diffusion.git synced 2026-04-16 13:49:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Apply suggestions from code review

Browse Source
This commit is contained in:
Patrick von Platen 2022-08-19 18:03:22 +02:00 committed by GitHub
parent d0c714ae4a
commit 536eb1a8ba
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/txt2img.py
View File

@ -237,9 +237,7 @@ def main():
if opt.fixed_code: if opt.fixed_code:
start_code = torch.randn([opt.n_samples, opt.C, opt.H // opt.f, opt.W // opt.f], device=device) start_code = torch.randn([opt.n_samples, opt.C, opt.H // opt.f, opt.W // opt.f], device=device)
print("start code", start_code.abs().sum())
precision_scope = autocast if opt.precision=="autocast" else nullcontext precision_scope = autocast if opt.precision=="autocast" else nullcontext
precision_scope = nullcontext
with torch.no_grad(): with torch.no_grad():
with precision_scope("cuda"): with precision_scope("cuda"):
with model.ema_scope(): with model.ema_scope():
@ -288,6 +286,7 @@ def main():
Image.fromarray(grid.astype(np.uint8)).save(os.path.join(outpath, f'grid-{grid_count:04}.png')) Image.fromarray(grid.astype(np.uint8)).save(os.path.join(outpath, f'grid-{grid_count:04}.png'))
grid_count += 1 grid_count += 1
toc = time.time()
image = x_samples_ddim.cpu().permute(0, 2, 3, 1).numpy() image = x_samples_ddim.cpu().permute(0, 2, 3, 1).numpy()
# run safety checker # run safety checker