|
|
import gradio as gr |
|
|
import modules.shared as shared |
|
|
from modules import scripts |
|
|
from scripts import lightdiffusionflow_config |
|
|
|
|
|
OutputPrompt = lightdiffusionflow_config.OutputPrompt |
|
|
|
|
|
def on_ui_settings(): |
|
|
|
|
|
section = ("lightdiffusionflow", "Light Diffusion Flow") |
|
|
|
|
|
|
|
|
|
|
|
shared.opts.add_option("lightdiffusionflow-language", |
|
|
shared.OptionInfo( |
|
|
"default", |
|
|
"显示语言/Language", |
|
|
gr.Dropdown, |
|
|
lambda: { |
|
|
"choices": ["default","中文","english"] |
|
|
}, section=section) |
|
|
) |
|
|
|
|
|
shared.opts.add_option("lightdiffusionflow-mode", |
|
|
shared.OptionInfo( |
|
|
"All", |
|
|
f"模式/Mode: ({OutputPrompt.note_for_save_mode()})", |
|
|
gr.Dropdown, |
|
|
lambda: { |
|
|
"choices": ["Core","All"] |
|
|
}, section=section) |
|
|
) |
|
|
|
|
|
shared.opts.add_option("lightdiffusionflow-local-flows-path", |
|
|
shared.OptionInfo( |
|
|
default="models/LightDiffusionFlow", |
|
|
label=f"本地保存的flow文件路径/Local path to save flow files", |
|
|
|
|
|
section=section) |
|
|
) |
|
|
|
|
|
shared.opts.add_option("lightdiffusionflow-auto-fix-params", |
|
|
shared.OptionInfo( |
|
|
True, |
|
|
f"自动纠正错误的参数/Automatically fix incorrect parameters", |
|
|
gr.Checkbox, |
|
|
{"interactive": True}, |
|
|
section=section) |
|
|
) |
|
|
|
|
|
|
|
|
scripts.script_callbacks.on_ui_settings(on_ui_settings) |
|
|
|