diff options
Diffstat (limited to 'infer.py')
-rw-r--r-- | infer.py | 5 |
1 files changed, 4 insertions, 1 deletions
@@ -262,7 +262,10 @@ def generate(output_dir, pipeline, args): | |||
262 | 262 | ||
263 | with autocast("cuda"): | 263 | with autocast("cuda"): |
264 | for i in range(args.batch_num): | 264 | for i in range(args.batch_num): |
265 | pipeline.set_progress_bar_config(desc=f"Batch {i + 1} of {args.batch_num}") | 265 | pipeline.set_progress_bar_config( |
266 | desc=f"Batch {i + 1} of {args.batch_num}", | ||
267 | dynamic_ncols=True | ||
268 | ) | ||
266 | 269 | ||
267 | generator = torch.Generator(device="cuda").manual_seed(seed + i) | 270 | generator = torch.Generator(device="cuda").manual_seed(seed + i) |
268 | images = pipeline( | 271 | images = pipeline( |