diff options
| author | Trung Ngo <codem01@gmail.com> | 2022-10-09 03:57:19 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-13 09:21:20 +0000 |
| commit | e72adc999b3531370eafb9d316924ac497feb445 (patch) | |
| tree | ea6f86009fa6b7f382d2b76ae51e67849bc86d76 /modules/generation_parameters_copypaste.py | |
| parent | 04c0e643f2eec68d93a76db171b4d70595808702 (diff) | |
| download | stable-diffusion-webui-gfx803-e72adc999b3531370eafb9d316924ac497feb445.tar.gz stable-diffusion-webui-gfx803-e72adc999b3531370eafb9d316924ac497feb445.tar.bz2 stable-diffusion-webui-gfx803-e72adc999b3531370eafb9d316924ac497feb445.zip | |
Restore last generation params
Diffstat (limited to 'modules/generation_parameters_copypaste.py')
| -rw-r--r-- | modules/generation_parameters_copypaste.py | 8 |
1 files changed, 8 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 = []
|