From 5e28ec34dabc625b38e7a967471f9d4dbda133ec Mon Sep 17 00:00:00 2001 From: dr413677671 <413677671@qq.com> Date: Sat, 28 Oct 2023 15:51:07 +0800 Subject: [PATCH] fix param --- paste_this_to_webui_scripts_folder/prompt_gallery.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/paste_this_to_webui_scripts_folder/prompt_gallery.py b/paste_this_to_webui_scripts_folder/prompt_gallery.py index 254da07..b5f4fd5 100644 --- a/paste_this_to_webui_scripts_folder/prompt_gallery.py +++ b/paste_this_to_webui_scripts_folder/prompt_gallery.py @@ -271,7 +271,8 @@ def add_param(key, value, cur_str): def parse_size(i_width, i_height, str_size, cur_str): i_width = str_size.split('x')[0] - i_height = str_size.split('x')[1] + i_height = str_size.split('x')[1] + return i_width, i_height def parse_virariant_size(str_size, cur_str): width = str_size.split('x')[0] @@ -306,7 +307,7 @@ def parse_param(param_str): elif key == 'Sampler': m_sampler_index = map_sampler_to_idx[value] elif key == 'Size': - parse_size(m_width, m_height, value, cur_line) + m_width, m_height = parse_size(m_width, m_height, value, cur_line) elif key == 'Seed resize from': cur_line = parse_virariant_size(value, cur_line) elif key == 'Seed':