Included seed in image name.

Beyondo · Apr 10, 2023 · 6fef7c9 · 6fef7c9
1 parent d66c494
commit 6fef7c9
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/legacy/img2img.py b/legacy/img2img.py
@@ -45,7 +45,7 @@ def process(ShouldSave, maxNumJobs, ShouldPreview = True, ReplaceResult = True):
                 callback_steps=20).images[0]
             colab.last_generated_image = image
             progress.show(image)
-            postprocessor.post_process(image, "%d_%d" % (timestamp, i), colab.get_current_image_uid(), ShouldSave, ReplaceResult)
+            postprocessor.post_process(image, f"{timestamp}_{i}_{colab.current_seed}", colab.get_current_image_uid(), ShouldSave, ReplaceResult)
             display(HTML("<label>Iterations: %d/%d</label>" % (i + 1,  num_iterations)), display_id="iterations")
         postprocessor.join_queue_thread()
         torch.cuda.empty_cache()

diff --git a/legacy/inpaint.py b/legacy/inpaint.py
@@ -69,7 +69,7 @@ def process(ShouldSave, maxNumJobs, ShouldPreview = True, ReplaceResult = True):
             image.thumbnail((512, 512))
             colab.last_generated_image = image
             progress.show(image)
-            postprocessor.post_process(image, "%d_%d" % (timestamp, i), colab.get_current_image_uid(), ShouldSave, ReplaceResult)
+            postprocessor.post_process(image, f"{timestamp}_{i}_{colab.current_seed}", colab.get_current_image_uid(), ShouldSave, ReplaceResult)
             display(HTML("<label>Iterations: %d/%d</label>" % (i + 1,  num_iterations)), display_id="iterations")
         postprocessor.join_queue_thread()
         torch.cuda.empty_cache()

diff --git a/legacy/text2img.py b/legacy/text2img.py
@@ -33,7 +33,7 @@ def process(ShouldSave, maxNumJobs, ShouldPreview = True, ReplaceResult = True):
                 callback_steps=20).images[0]
             colab.last_generated_image = image
             progress.show(image)
-            postprocessor.post_process(image, "%d_%d" % (timestamp, i), colab.get_current_image_uid(), ShouldSave, ReplaceResult)
+            postprocessor.post_process(image, f"{timestamp}_{i}_{colab.current_seed}", colab.get_current_image_uid(), ShouldSave, ReplaceResult)
             display(HTML("<label>Iterations: %d/%d</label>" % (i + 1,  num_iterations)), display_id="iterations")
         postprocessor.join_queue_thread()
         torch.cuda.empty_cache()