From f16ea46cce5b7e2ddfb76a64b8f342c03d8a956a Mon Sep 17 00:00:00 2001
From: AUTOMATIC1111 <16777216c@gmail.com>
Date: Thu, 22 Sep 2022 22:29:52 +0300
Subject: [PATCH] Updated Custom Scripts (markdown)

---
 Custom-Scripts.md | 41 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)

diff --git a/Custom-Scripts.md b/Custom-Scripts.md
index ad86d9a..0bb879b 100644
--- a/Custom-Scripts.md
+++ b/Custom-Scripts.md
@@ -146,4 +146,45 @@ class Script(scripts.Script):
             proc.seed + i, proc.prompt, opts.samples_format, info= proc.info, p=p)
 
         return proc
+```
+
+
+# Saving steps of the sampling process
+This script will save steps of the sampling process to a directory.
+```
+import os.path
+
+import modules.scripts as scripts
+import gradio as gr
+
+from modules import sd_samplers, shared
+from modules.processing import Processed, process_images
+
+
+class Script(scripts.Script):
+    def title(self):
+        return "Save steps of the sampling process to files"
+
+    def ui(self, is_img2img):
+        path = gr.Textbox(label="Save images to path")
+        return [path]
+
+    def run(self, p, path):
+        index = [0]
+
+        def store_latent(x):
+            image = shared.state.current_image = sd_samplers.sample_to_image(x)
+            image.save(os.path.join(path, f"sample-{index[0]:05}.png"))
+            index[0] += 1
+            fun(x)
+
+        fun = sd_samplers.store_latent
+        sd_samplers.store_latent = store_latent
+
+        try:
+            proc = process_images(p)
+        finally:
+            sd_samplers.store_latent = fun
+
+        return Processed(p, proc.images, p.seed, "")
 ```
\ No newline at end of file