diff options
Diffstat (limited to 'scripts')
| -rw-r--r-- | scripts/custom_code.py | 6 | ||||
| -rw-r--r-- | scripts/loopback.py | 6 |
2 files changed, 8 insertions, 4 deletions
diff --git a/scripts/custom_code.py b/scripts/custom_code.py index 5694f2dd..a9b10c09 100644 --- a/scripts/custom_code.py +++ b/scripts/custom_code.py @@ -4,8 +4,8 @@ import gradio as gr from modules.processing import Processed
from modules.shared import opts, cmd_opts, state
-
class Script(scripts.Script):
+
def title(self):
return "Custom code"
@@ -18,6 +18,7 @@ class Script(scripts.Script): return [code]
+
def run(self, p, code):
assert cmd_opts.allow_code, '--allow-code option must be enabled'
@@ -37,4 +38,5 @@ class Script(scripts.Script): exec(compiled, module.__dict__)
return Processed(p, *display_result_data)
-
+
+
\ No newline at end of file diff --git a/scripts/loopback.py b/scripts/loopback.py index 8aca61f3..e90b58d4 100644 --- a/scripts/loopback.py +++ b/scripts/loopback.py @@ -40,8 +40,7 @@ class Script(scripts.Script): all_images = []
state.job_count = loops * batch_count
- if opts.img2img_color_correction:
- p.color_corrections = [processing.setup_color_correction(p.init_images[0])]
+ initial_color_corrections = [processing.setup_color_correction(p.init_images[0])]
for n in range(batch_count):
history = []
@@ -51,6 +50,9 @@ class Script(scripts.Script): p.batch_size = 1
p.do_not_save_grid = True
+ if opts.img2img_color_correction:
+ p.color_corrections = initial_color_corrections
+
state.job = f"Iteration {i + 1}/{loops}, batch {n + 1}/{batch_count}"
processed = processing.process_images(p)
|