From ba4892e03fcfab4c8dac71c453cee45f2d4f6568 Mon Sep 17 00:00:00 2001
From: Kevin Schaul <kevin.schaul@gmail.com>
Date: Fri, 16 Sep 2022 16:32:18 -0500
Subject: [PATCH] Zero-pad intermediate image file names (#616)

---
 ldm/dream/server.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ldm/dream/server.py b/ldm/dream/server.py
index cde3957a1f..372d719052 100644
--- a/ldm/dream/server.py
+++ b/ldm/dream/server.py
@@ -228,7 +228,8 @@ class DreamServer(BaseHTTPRequestHandler):
             nonlocal step_index
             if opt.progress_images and step % 5 == 0 and step < opt.steps - 1:
                 image = self.model.sample_to_image(sample)
-                name = f'{prefix}.{opt.seed}.{step_index}.png'
+                step_index_padded = str(step_index).rjust(len(str(opt.steps)), '0')
+                name = f'{prefix}.{opt.seed}.{step_index_padded}.png'
                 metadata = f'{opt.prompt} -S{opt.seed} [intermediate]'
                 path = step_writer.save_image_and_prompt_to_png(image, dream_prompt=metadata, name=name)
                 step_index += 1