Addes support for tailed diffusion

javascript/state.core.js

@@ -48,6 +48,7 @@ state.core = (function () {
     const TOGGLE_BUTTONS = { // Toggle buttons
         'hires_fix': 'hr',
         'refiner': 'enable',
+        'tiled_diffusion': 'MD-t2i-enabled',
     };
 
     // *** Exceptions *** //
@@ -63,6 +64,7 @@ state.core = (function () {
         'upscaler_scale_to_crop',
         'upscaler_1',
         'upscaler_2',
+        'tiled_diffusion',
     ];
 
     // Elements that have the same id in different tabs
@@ -75,6 +77,7 @@ state.core = (function () {
         'upscaler_scale_to_crop',
         'upscaler_1',
         'upscaler_2',
+        'tiled_diffusion',
     ];
 
     let store = null;
@@ -349,7 +352,7 @@ state.core = (function () {
     function handleToggleButton(id, duplicateIds) {
         let btn = gradioApp().querySelector(`button#${id}`);
         if (! btn) { // New gradio version
-            btn = gradioApp().querySelector(`.input-accordion#${id}`);
+            btn = gradioApp().querySelector(`.input-accordion#${id} .label-wrap`);
         }
         if (! btn) {
             state.logging.warn(`Button not found: ${id}`);
@@ -359,7 +362,7 @@ state.core = (function () {
             state.utils.clickToggleMenu(btn);
         }
         btn.addEventListener('click', function () {
-            let classList = Array.from(this.classList);
+            let classList = Array.from(this.parentNode.classList);
             if (btn.tagName === 'BUTTON') { // Old gradio version
                 store.set(id, classList.indexOf('secondary-down') === -1);
             } else {

javascript/state.utils.js

@@ -35,12 +35,13 @@ state.utils = {
             cancelable: true
         });
         element.dispatchEvent(mouseEvent);
-        let icon = element.querySelector('.icon');
+        let icon = element.parentNode.querySelector('.icon');
         if (icon) {
             icon.dispatchEvent(mouseEvent);
         }
-        let checkbox = element.querySelector('input[type="checkbox"]');
+        let checkbox = element.parentNode.querySelector('input[type="checkbox"]');
         if (checkbox) {
+            checkbox.checked = !checkbox.checked;
             checkbox.dispatchEvent(mouseEvent);
         }
         return element;

scripts/state_settings.py

@@ -45,6 +45,7 @@ def on_ui_settings():
             "upscaler_1",
             "upscaler_2",
             "upscaler_2_visibility",
+            "tiled_diffusion",
             "script"
         ]
     }, section=section))
@@ -76,6 +77,7 @@ def on_ui_settings():
             "cfg_scale",
             "denoising_strength",
             "seed",
+            "tiled_diffusion",
             "script"
         ]
     }, section=section))