kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

web_config add sd_local_model_path

Browse Source
This commit is contained in:
Qing 2023-03-31 09:09:37 +08:00
parent b5dfa06dfa
commit d3aaed5af8
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
lama_cleaner/web_config.py
View File

@ -15,6 +15,7 @@ class Config(BaseModel):
host: str = "127.0.0.1" host: str = "127.0.0.1"
port: int = 8080 port: int = 8080
model: str = DEFAULT_MODEL model: str = DEFAULT_MODEL
sd_local_model_path: str = None
sd_controlnet: bool = False sd_controlnet: bool = False
device: str = DEFAULT_DEVICE device: str = DEFAULT_DEVICE
gui: bool = False gui: bool = False
@ -53,6 +54,7 @@ def save_config(
host, host,
port, port,
model, model,
sd_local_model_path,
sd_controlnet, sd_controlnet,
device, device,
gui, gui,
@ -195,6 +197,9 @@ def main(config_file: str):
enable_gif = gr.Checkbox(init_config.enable_gif, label=GIF_HELP) enable_gif = gr.Checkbox(init_config.enable_gif, label=GIF_HELP)
with gr.Tab("Diffusion Model"): with gr.Tab("Diffusion Model"):
sd_local_model_path = gr.Checkbox(
init_config.sd_local_model_path, label=f"{SD_LOCAL_MODEL_HELP}"
)
sd_controlnet = gr.Checkbox( sd_controlnet = gr.Checkbox(
init_config.sd_controlnet, label=f"{SD_CONTROLNET_HELP}" init_config.sd_controlnet, label=f"{SD_CONTROLNET_HELP}"
) )
@ -202,12 +207,12 @@ def main(config_file: str):
cpu_offload = gr.Checkbox( cpu_offload = gr.Checkbox(
init_config.cpu_offload, label=f"{CPU_OFFLOAD_HELP}" init_config.cpu_offload, label=f"{CPU_OFFLOAD_HELP}"
) )
disable_nsfw = gr.Checkbox(
init_config.disable_nsfw, label=f"{DISABLE_NSFW_HELP}"
)
sd_cpu_textencoder = gr.Checkbox( sd_cpu_textencoder = gr.Checkbox(
init_config.sd_cpu_textencoder, label=f"{SD_CPU_TEXTENCODER_HELP}" init_config.sd_cpu_textencoder, label=f"{SD_CPU_TEXTENCODER_HELP}"
) )
disable_nsfw = gr.Checkbox(
init_config.disable_nsfw, label=f"{DISABLE_NSFW_HELP}"
)
enable_xformers = gr.Checkbox( enable_xformers = gr.Checkbox(
init_config.enable_xformers, label=f"{ENABLE_XFORMERS_HELP}" init_config.enable_xformers, label=f"{ENABLE_XFORMERS_HELP}"
) )
@ -221,6 +226,7 @@ def main(config_file: str):
host, host,
port, port,
model, model,
sd_local_model_path,
sd_controlnet, sd_controlnet,
device, device,
gui, gui,