From 63ef7b09fbb41af56211e2fb0f514f7f38ebf275 Mon Sep 17 00:00:00 2001 From: Zyin055 <5003647+Zyin055@users.noreply.github.com> Date: Mon, 2 Jan 2023 21:45:40 -0600 Subject: [PATCH] fix sampler not being saved for new presets --- scripts/config_presets.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/scripts/config_presets.py b/scripts/config_presets.py index 222a18d..8334653 100644 --- a/scripts/config_presets.py +++ b/scripts/config_presets.py @@ -1,4 +1,5 @@ import traceback +import modules.sd_samplers import modules.scripts as scripts import gradio as gr import json @@ -447,7 +448,15 @@ def save_config(config_presets, component_map, config_file_name): for i, component_id in enumerate(component_map.keys()): if component_map[component_id] is not None: - new_setting_map[component_id] = new_setting[i] + new_value = new_setting[i] # this gives the index when the component is a dropdown + if component_id == "txt2img_sampling": + new_setting_map[component_id] = modules.sd_samplers.samplers[new_value].name + elif component_id == "img2img_sampling": + new_setting_map[component_id] = modules.sd_samplers.samplers_for_img2img[new_value].name + else: + new_setting_map[component_id] = new_value + + #print(f"new_setting_map = {new_setting_map}") config_presets.update({new_setting_name: new_setting_map}) write_config_presets_to_file(config_presets, config_file_name)