tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

スタイルロード時にデフォルトですべての組み合わせが入るようにした

Browse Source
This commit is contained in:
frodo821
2024-02-27 04:39:09 +09:00
parent d0b5bc8682
commit a48e5cbc51
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
webui_merge.py
View File

@@ -268,7 +268,12 @@ def load_styles_gr(model_name_a, model_name_b):
with open(config_path_b, encoding="utf-8") as f: with open(config_path_b, encoding="utf-8") as f:
config_b = json.load(f) config_b = json.load(f)
styles_b = list(config_b["data"]["style2id"].keys()) styles_b = list(config_b["data"]["style2id"].keys())
return gr.Textbox(value=", ".join(styles_a)), gr.Textbox(value=", ".join(styles_b))
return gr.Textbox(value=", ".join(styles_a)), gr.Textbox(value=", ".join(styles_b)), gr.TextArea(
label="スタイルのマージリスト",
placeholder=f"{DEFAULT_STYLE}, {DEFAULT_STYLE},{DEFAULT_STYLE}\nAngry, Angry, Angry",
value='\n'.join(f"{sty_a}, {sty_b}, {sty_a if sty_a != sty_b else ''}{sty_b}" for sty_a in styles_a for sty_b in styles_b),
)
initial_md = """ initial_md = """
@@ -440,7 +445,7 @@ with gr.Blocks(theme=GRADIO_THEME) as app:
load_style_button.click( load_style_button.click(
load_styles_gr, load_styles_gr,
inputs=[model_name_a, model_name_b], inputs=[model_name_a, model_name_b],
outputs=[styles_a, styles_b], outputs=[styles_a, styles_b, style_triple_list],
) )
model_merge_button.click( model_merge_button.click(