Add steps/second info to log message

This commit is contained in:
oobabooga 2025-12-01 15:44:31 -08:00
parent 6a7209a842
commit 748e2e55fd

View file

@ -643,7 +643,7 @@ def generate(state):
t1 = time.time()
save_generated_images(all_images, state, seed)
logger.info(f'Images generated in {(t1-t0):.2f} seconds (seed {seed})')
logger.info(f'Images generated in {(t1-t0):.2f} seconds ({state["image_steps"]/(t1-t0):.2f} steps/s, seed {seed})')
return all_images