From 536eb1a8ba9e7e3806727cc25331d43b1f076f07 Mon Sep 17 00:00:00 2001
From: Patrick von Platen <patrick.v.platen@gmail.com>
Date: Fri, 19 Aug 2022 18:03:22 +0200
Subject: [PATCH] Apply suggestions from code review
---
scripts/txt2img.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/scripts/txt2img.py b/scripts/txt2img.py
index 0af430c..e93cb19 100644
--- a/scripts/txt2img.py
+++ b/scripts/txt2img.py
@@ -237,9 +237,7 @@ def main():
if opt.fixed_code:
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 = nullcontext
with torch.no_grad():
with precision_scope("cuda"):
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'))
grid_count += 1
+ toc = time.time()
image = x_samples_ddim.cpu().permute(0, 2, 3, 1).numpy()
# run safety checker
--
GitLab