From 02f52b9dc1771bca3e5a6d0c087a2fd6f1be8774 Mon Sep 17 00:00:00 2001
From: "ilian.iliev" <ilian.iliev@imperiaonline.org>
Date: Fri, 3 May 2024 21:32:48 +0300
Subject: [PATCH] Added refiner elements state

---
 javascript/state.core.js  | 7 +++++--
 scripts/state_settings.py | 6 ++++++
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/javascript/state.core.js b/javascript/state.core.js
index 7a23065..941b15a 100644
--- a/javascript/state.core.js
+++ b/javascript/state.core.js
@@ -15,13 +15,14 @@ state.core = (function () {
         'hires_resize_x': 'hr_resize_x',
         'hires_resize_y': 'hr_resize_y',
         'hires_denoising_strength': 'denoising_strength',
+        'refiner_switch': 'switch_at',
         'width': 'width',
         'height': 'height',
         'batch_count': 'batch_count',
         'batch_size': 'batch_size',
         'cfg_scale': 'cfg_scale',
         'denoising_strength': 'denoising_strength',
-        'seed': 'seed'
+        'seed': 'seed',
     };
 
     const ELEMENTS_WITHOUT_PREFIX = {
@@ -32,6 +33,7 @@ state.core = (function () {
         'sampling': 'sampling',
         'scheduler': 'scheduler',
         'hires_upscaler': 'hr_upscaler',
+        'refiner_checkpoint': 'checkpoint',
         'script': '#script_list',
     };
 
@@ -41,7 +43,8 @@ state.core = (function () {
 
     const TOGGLE_BUTTONS = {
         'extra_networks': 'extra_networks',
-        'hires_fix': 'hr'
+        'hires_fix': 'hr',
+        'refiner': 'enable',
     };
 
     let store = null;
diff --git a/scripts/state_settings.py b/scripts/state_settings.py
index 1da417c..9efe5c1 100644
--- a/scripts/state_settings.py
+++ b/scripts/state_settings.py
@@ -37,6 +37,9 @@ def on_ui_settings():
             "hires_resize_x",
             "hires_resize_y",
             "hires_denoising_strength",
+            "refiner",
+            "refiner_checkpoint",
+            "refiner_switch",
             "script"
         ]
     }, section=section))
@@ -47,6 +50,9 @@ def on_ui_settings():
             "negative_prompt",
             "extra_networks",
             "styles",
+            "refiner",
+            "refiner_checkpoint",
+            "refiner_switch",
             "sampling",
             "scheduler",
             "resize_mode",