Restore last generation params

2 files changed, 12 insertions, 0 deletions

diff --git a/modules/generation_parameters_copypaste.py b/modules/generation_parameters_copypaste.py
index ac1ba7f4..3e75aecc 100644
--- a/modules/generation_parameters_copypaste.py
+++ b/modules/generation_parameters_copypaste.py
@@ -1,5 +1,7 @@
+import os

 import re

 import gradio as gr

+from modules.shared import script_path

 

 re_param_code = r"\s*([\w ]+):\s*([^,]+)(?:,|$)"

 re_param = re.compile(re_param_code)

@@ -61,6 +63,12 @@ Steps: 20, Sampler: Euler a, CFG scale: 7, Seed: 965400086, Size: 512x512, Model
 

 def connect_paste(button, paste_fields, input_comp, js=None):

     def paste_func(prompt):

+        if not prompt:

+            filename = os.path.join(script_path, "params.txt")

+            if os.path.exists(filename):

+                with open(filename, "r", encoding="utf8") as file:

+                    prompt = file.read()

+

         params = parse_generation_parameters(prompt)

         res = []

 

diff --git a/modules/processing.py b/modules/processing.py
index 698b3069..d5172f00 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -324,6 +324,10 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
     else:

         assert p.prompt is not None

 

+    with open(os.path.join(shared.script_path, "params.txt"), "w", encoding="utf8") as file:

+        processed = Processed(p, [], p.seed, "")

+        file.write(processed.infotext(p, 0))

+

     devices.torch_gc()

 

     seed = get_fixed_seed(p.seed)