mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2026-01-18 14:40:53 +01:00
Improve a log message
This commit is contained in:
parent
180e1f0cbf
commit
f41e558e00
|
|
@ -732,7 +732,7 @@ def generate(state):
|
|||
|
||||
total_images = int(state['image_batch_count']) * int(state['image_batch_size'])
|
||||
total_steps = state["image_steps"] * int(state['image_batch_count'])
|
||||
logger.info(f'Generated {total_images} images in {(t1 - t0):.2f} seconds ({total_steps / (t1 - t0):.2f} steps/s, seed {seed})')
|
||||
logger.info(f'Generated {total_images} {"image" if total_images == 1 else "images"} in {(t1 - t0):.2f} seconds ({total_steps / (t1 - t0):.2f} steps/s, seed {seed})')
|
||||
|
||||
return all_images
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue