Zyin055/Config-Presets
1
0
Fork 0
mirror of https://github.com/Zyin055/Config-Presets.git synced 2026-01-26 19:19:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor, made it work with img2img

Browse Source 
Removed ast and replaced with json,
ast uses eval, not safe

Used a dictionary as a map and a Constant
as a settings area
This commit is contained in:
Gerschel
2022-12-13 03:01:23 -08:00
parent 258c347a0c
commit 318de88d00
2 changed files with 99 additions and 108 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
config.json
View File

@@ -6,54 +6,54 @@
"sampler_index": "DPM++ 2M Karras",
"width": 512,
"height": 512,
"enable_hr": False,
"enable_hr": false,
"denoising_strength": 0.7,
"batch_count": 1,
"batch_size": 8,
"cfg_scale": 7,
"cfg_scale": 7
},
"Medium quality - 512x512, steps: 20, batch size: 8, DPM++ 2S a Karras" : {
"steps": 20,
"sampler_index": "DPM++ 2S a Karras",
"width": 512,
"height": 512,
"enable_hr": False,
"enable_hr": false,
"denoising_strength": 0.7,
"batch_count": 1,
"batch_size": 8,
"cfg_scale": 7,
"cfg_scale": 7
},
"High quality ------ 512x512, steps: 40, batch size: 8, DPM++ 2S a Karras" : {
"steps": 40,
"sampler_index": "DPM++ 2S a Karras",
"width": 512,
"height": 512,
"enable_hr": False,
"enable_hr": false,
"denoising_strength": 0.7,
"batch_count": 1,
"batch_size": 8,
"cfg_scale": 7,
"cfg_scale": 7
},
"High res -------- 1024x1024, steps: 30, batch size: 1, DPM++ 2M Karras, Highres fix, Denoising: 0.4" : {
"steps": 30,
"sampler_index": "DPM++ 2M Karras",
"width": 1024,
"height": 1024,
"enable_hr": True,
"enable_hr": true,
"denoising_strength": 0.4,
"batch_count": 1,
"batch_size": 1,
"cfg_scale": 7,
"cfg_scale": 7
},
"Wallpaper ----- 1920x1088, steps: 30, batch size: 1, DPM++ 2M Karras, Highres fix, Denoising: 0.3" : {
"steps": 30,
"sampler_index": "DPM++ 2M Karras",
"width": 1920,
"height": 1088,
"enable_hr": True,
"enable_hr": true,
"denoising_strength": 0.3,
"batch_count": 1,
"batch_size": 1,
"cfg_scale": 7,
},
"cfg_scale": 7
}
}

185
scripts/config_presets.py
View File

@@ -1,40 +1,49 @@
# Config Presets script for Automatic1111
# Last updated: 12/13/2022
# Author: Zyin
import modules.scripts
import gradio as gr
import ast
import json
import os
import platform
import subprocess as sp
ui_steps = None # "steps" component in ui.py:635, label='Sampling Steps'
ui_sampler_index = None # "steps" component in ui.py:636, label='Sampling method'
ui_width = None # "width" component in ui.py:639, label='Width'
ui_height = None # "height" component in ui.py:640, label='Height'
ui_enable_hr = None # "enable_hr" component in ui.py:645,label='Highres. fix'
ui_denoising_strength = None # "denoising_strength" component in ui.py:650, label='Denoising strength'
ui_batch_count = None # "batch_count" component in ui.py:653, label='Batch count'
ui_batch_size = None # "batch_size" component in ui.py:654, label='Batch size'
ui_cfg_scale = None # "cfg_scale" component in ui.py:656, label='CFG Scale'
#print(f"global modules.scripts.basedir()={modules.scripts.basedir()}")
basedir = modules.scripts.basedir() #C:\Stable Diffusion\extensions\Config-Presets
# Load config from file
config_presets = {}
try:
with open(f"{basedir}/config.json") as file:
config_presets = ast.literal_eval(file.read())
#print("Config Presets: loaded config file 'config.json'")
except FileNotFoundError:
print(f"[ERROR] Config Presets: Could not find config file at '{basedir}/config.json'. The Config Presets dropdown will not work!")
#Acts like settings for components
AVAILABLE_COMPONENTS = [
"Sampling Steps",
"Sampling method",
"Width",
"Height",
"Highres. fix",
"Denoising strength",
"Batch count",
"Batch size",
"CFG Scale",
"Seed",
]
#ui_steps = None # "steps" component in ui.py:635, label='Sampling Steps'
#ui_sampler_index = None # "steps" component in ui.py:636, label='Sampling method'
#ui_width = None # "width" component in ui.py:639, label='Width'
#ui_height = None # "height" component in ui.py:640, label='Height'
#ui_enable_hr = None # "enable_hr" component in ui.py:645,label='Highres. fix'
#ui_denoising_strength = None # "denoising_strength" component in ui.py:650, label='Denoising strength'
#ui_batch_count = None # "batch_count" component in ui.py:653, label='Batch count'
#ui_batch_size = None # "batch_size" component in ui.py:654, label='Batch size'
#ui_cfg_scale = None # "cfg_scale" component in ui.py:656, label='CFG Scale'
class Script(modules.scripts.Script):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.component_map = {k:None for k in AVAILABLE_COMPONENTS}
self.basedir = modules.scripts.basedir() #C:\Stable Diffusion\extensions\Config-Presets
#Load config from file
try:
with open(f"{self.basedir}/extensions\\Config-Presets\\config.json") as file:
self.config_presets = json.load(file)
except FileNotFoundError:
print(f"[ERROR] Config Presets: Could not find config file at '{self.basedir}/extensions\\Config-Presets\\config.json'. The Config Presets dropdown will not work!")
def title(self):
return "Config Presets"
@@ -43,86 +52,57 @@ class Script(modules.scripts.Script):
return modules.scripts.AlwaysVisible # hide this script in the Scripts dropdown
def after_component(self, component, **kwargs):
global ui_steps
global ui_sampler_index
global ui_width
global ui_height
global ui_enable_hr
global ui_denoising_strength
global ui_batch_count
global ui_batch_size
global ui_cfg_scale
# Since we can't get direct access to component variables in ui.py, we detect and save an old reference to them here.
if isinstance(component, gr.components.Slider):
if component.label == "Sampling Steps":
ui_steps = component
#print(f"found ui_steps component: {type(component)}")
elif component.label == "Width":
ui_width = component
#print(f"found ui_width component: {type(component)}")
elif component.label == "Height":
ui_height = component
#print(f"found ui_height component: {type(component)}")
elif component.label == "Denoising strength":
ui_denoising_strength = component
#print(f"found ui_denoising_strength component: {type(component)}")
elif component.label == "Batch count":
ui_batch_count = component
#print(f"found ui_batch_count component: {type(component)}")
elif component.label == "Batch size":
ui_batch_size = component
#print(f"found ui_batch_size component: {type(component)}")
elif component.label == "CFG Scale":
ui_cfg_scale = component
#print(f"found ui_cfg_scale component: {type(component)}")
elif isinstance(component, gr.components.Checkbox):
if component.label == "Highres. fix":
ui_enable_hr = component
#print(f"found ui_enable_hr component: {type(component)}")
elif isinstance(component, gr.components.Radio):
if component.label == "Sampling method":
ui_sampler_index = component
#print(f"found ui_sampler_index component: {type(component)}")
# Getting tricky using try block instead of `if label in dict`
try:
self.component_map[component.label]
self.component_map[component.label] = component
except KeyError:
# Not the component we're looking for, not a real error, pass
pass
#if component.elem_id == "script_list": #bottom of the script dropdown
#if component.elem_id == "txt2img_style2_index": #doesn't work, need to be added after all the components we edit are loaded
if component.elem_id == "open_folder": #bottom of the image gallery
global config_presets
preset_values = []
for k in config_presets:
for k in self.config_presets:
preset_values.append(k)
#print(f"Config Presets: added \"{k}\"")
with gr.Column(scale=9):
def config_preset_dropdown_change(dropdown_value):
global config_presets
config_preset = config_presets[dropdown_value]
config_preset = self.config_presets[dropdown_value]
print(f"Config Presets: changed to {dropdown_value}")
# return config_dict["steps"] if "steps" in config_dict else 20,\
# config_dict["sampler_index"] if "sampler_index" in config_dict else "Euler a",\
# config_dict["width"] if "width" in config_dict else 512,\
# config_dict["height"] if "height" in config_dict else 512,\
# config_dict["enable_hr"] if "enable_hr" in config_dict else False,\
# config_dict["denoising_strength"] if "denoising_strength" in config_dict else 0.7,\
# config_dict["batch_count"] if "batch_count" in config_dict else 1,\
# config_dict["batch_size"] if "batch_size" in config_dict else 1,\
# config_dict["cfg_scale"] if "cfg_scale" in config_dict else 7,\
# I couldn't find a way to get up-to-date values from the UI elements, so instead we just use the default values if something is missing from the config file
if self.component_map["Highres. fix"]:
#if we are text2img highres fix gets a component, if it's None still, continue
return (config_preset["steps"] if "steps" in config_preset else self.component_map["Sampling Steps"].value,
config_preset["sampler_index"] if "sampler_index" in config_preset else self.component_map["Sampling method"].value,
config_preset["width"] if "width" in config_preset else self.component_map["Width"].value,
config_preset["height"] if "height" in config_preset else self.component_map["Height"].value,
config_preset["enable_hr"] if "enable_hr" in config_preset else self.component_map["Highres. fix"].value,
config_preset["denoising_strength"] if "denoising_strength" in config_preset else self.component_map["Denoising strength"].value,
config_preset["batch_count"] if "batch_count" in config_preset else self.component_map["Batch count"].value,
config_preset["batch_size"] if "batch_size" in config_preset else self.component_map["Batch size"].value,
config_preset["cfg_scale"] if "cfg_scale" in config_preset else self.component_map["CFG Scale"].value
)
else:
# if we are img2img highres fix component is empty
return (config_preset["steps"] if "steps" in config_preset else self.component_map["Sampling Steps"].value,
config_preset["sampler_index"] if "sampler_index" in config_preset else self.component_map["Sampling method"].value,
config_preset["width"] if "width" in config_preset else self.component_map["Width"].value,
config_preset["height"] if "height" in config_preset else self.component_map["Height"].value,
#config_preset["enable_hr"] if "enable_hr" in config_preset else self.component_map["Highres. fix"].value,
config_preset["denoising_strength"] if "denoising_strength" in config_preset else self.component_map["Denoising strength"].value,
config_preset["batch_count"] if "batch_count" in config_preset else self.component_map["Batch count"].value,
config_preset["batch_size"] if "batch_size" in config_preset else self.component_map["Batch size"].value,
config_preset["cfg_scale"] if "cfg_scale" in config_preset else self.component_map["CFG Scale"].value
)
return config_preset["steps"] if "steps" in config_preset else ui_steps.value,\
config_preset["sampler_index"] if "sampler_index" in config_preset else ui_sampler_index.value,\
config_preset["width"] if "width" in config_preset else ui_width.value,\
config_preset["height"] if "height" in config_preset else ui_height.value,\
config_preset["enable_hr"] if "enable_hr" in config_preset else ui_enable_hr.value,\
config_preset["denoising_strength"] if "denoising_strength" in config_preset else ui_denoising_strength.value,\
config_preset["batch_count"] if "batch_count" in config_preset else ui_batch_count.value,\
config_preset["batch_size"] if "batch_size" in config_preset else ui_batch_size.value,\
config_preset["cfg_scale"] if "cfg_scale" in config_preset else ui_cfg_scale.value,\
config_preset_dropdown = gr.Dropdown(
label="Config Presets",
@@ -130,12 +110,23 @@ class Script(modules.scripts.Script):
elem_id="config_preset_dropdown",
)
config_preset_dropdown.style(container=False) #set to True to give it a white box to sit in
config_preset_dropdown.change(
fn=config_preset_dropdown_change,
show_progress=False,
inputs=[config_preset_dropdown],
outputs=[ui_steps, ui_sampler_index, ui_width, ui_height, ui_enable_hr, ui_denoising_strength, ui_batch_count, ui_batch_size, ui_cfg_scale],
)
if self.component_map["Highres. fix"]:
config_preset_dropdown.change(
fn=config_preset_dropdown_change,
show_progress=False,
inputs=[config_preset_dropdown],
#outputs=[ui_steps, ui_sampler_index, ui_width, ui_height, ui_enable_hr, ui_denoising_strength, ui_batch_count, ui_batch_size, ui_cfg_scale],
outputs = [self.component_map["Sampling Steps"],self.component_map["Sampling method"],self.component_map["Width"],self.component_map["Height"],self.component_map["Highres. fix"],self.component_map["Denoising strength"],self.component_map["Batch count"],self.component_map["Batch size"],self.component_map["CFG Scale"]]
)
else:
config_preset_dropdown.change(
fn=config_preset_dropdown_change,
show_progress=False,
inputs=[config_preset_dropdown],
#outputs = list([self.component_map[e] for e in AVAILABLE_COMPONENTS if e != "Seeds" and e != "Highres. fix"]) # **** LIST COMPS FAIL W/ GRADIO'S IN/OUTPUTS
outputs = [self.component_map["Sampling Steps"],self.component_map["Sampling method"],self.component_map["Width"],self.component_map["Height"],self.component_map["Denoising strength"],self.component_map["Batch count"],self.component_map["Batch size"],self.component_map["CFG Scale"]]
)
config_preset_dropdown.change(
fn=None,
inputs=[],
@@ -166,7 +157,7 @@ class Script(modules.scripts.Script):
elem_id="config_presets_open_config_file_button"
)
open_config_file_button.click(
fn=lambda: open_file(f"{basedir}\\config.json"),
fn=lambda: open_file(f"{self.basedir}extensions\\Config-Presets\\config.json"),
inputs=[],
outputs=[],
)