From 3ba59df559ea497107e0fcbaaa97a396e5a6cf9c Mon Sep 17 00:00:00 2001
From: Manuel Schmid <dev@mash1t.de>
Date: Sun, 24 Dec 2023 23:31:46 +0100
Subject: [PATCH] add default_overwrite_step handling for meta data and gradio

allows turbo preset switching to set default_overwrite_step correctly
---
 modules/async_worker.py | 1 +
 modules/config.py       | 1 +
 modules/meta_parser.py  | 8 ++++++++
 webui.py                | 2 ++
 4 files changed, 12 insertions(+)

diff --git a/modules/async_worker.py b/modules/async_worker.py
index 59c818cc..163786a8 100644
--- a/modules/async_worker.py
+++ b/modules/async_worker.py
@@ -792,6 +792,7 @@ def worker():
                         ('Refiner Switch', refiner_switch),
                         ('Sampler', sampler_name),
                         ('Scheduler', scheduler_name),
+                        ('Sampling Steps Override', advanced_parameters.overwrite_step),
                         ('Seed', task['task_seed']),
                     ]
                     for li, (n, w) in enumerate(loras):
diff --git a/modules/config.py b/modules/config.py
index 9f4b7aba..7baa0807 100644
--- a/modules/config.py
+++ b/modules/config.py
@@ -379,6 +379,7 @@ possible_preset_keys = {
     "checkpoint_downloads": "checkpoint_downloads",
     "embeddings_downloads": "embeddings_downloads",
     "lora_downloads": "lora_downloads",
+    "default_overwrite_step": "Sampling Steps Override"
 }
 
 
diff --git a/modules/meta_parser.py b/modules/meta_parser.py
index ebd72eb4..636c1c49 100644
--- a/modules/meta_parser.py
+++ b/modules/meta_parser.py
@@ -118,6 +118,14 @@ def load_parameter_button_click(raw_prompt_txt):
     except:
         results.append(gr.update())
 
+    try:
+        h = loaded_parameter_dict.get('Sampling Steps Override', None)
+        assert h is not None
+        h = float(h)
+        results.append(h)
+    except:
+        results.append(gr.update())
+
     try:
         h = loaded_parameter_dict.get('Seed', None)
         assert h is not None
diff --git a/webui.py b/webui.py
index 576d9b6a..6daafb46 100644
--- a/webui.py
+++ b/webui.py
@@ -536,6 +536,7 @@ with shared.gradio_root:
             refiner_switch,
             sampler_name,
             scheduler_name,
+            overwrite_step,
             seed_random,
             image_seed,
             generate_button,
@@ -641,6 +642,7 @@ with shared.gradio_root:
             refiner_switch,
             sampler_name,
             scheduler_name,
+            overwrite_step,
             seed_random,
             image_seed,
             generate_button,