You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
from typing import List
|
|
import gradio as gr
|
|
from modules import sd_models, sd_samplers
|
|
from scripts.faceswaplab_utils.sd_utils import get_sd_option
|
|
|
|
|
|
def face_inpainting_ui(id_prefix: str = "faceswaplab") -> List[gr.components.Component]:
|
|
inpainting_denoising_strength = gr.Slider(
|
|
0,
|
|
1,
|
|
0,
|
|
step=0.01,
|
|
elem_id=f"{id_prefix}_pp_inpainting_denoising_strength",
|
|
label="Denoising strenght",
|
|
)
|
|
|
|
inpainting_denoising_prompt = gr.Textbox(
|
|
get_sd_option(
|
|
"faceswaplab_pp_default_inpainting_prompt", "Portrait of a [gender]"
|
|
),
|
|
elem_id=f"{id_prefix}_pp_inpainting_denoising_prompt",
|
|
label="Inpainting prompt use [gender] instead of men or woman",
|
|
)
|
|
inpainting_denoising_negative_prompt = gr.Textbox(
|
|
get_sd_option("faceswaplab_pp_default_inpainting_negative_prompt", "blurry"),
|
|
elem_id=f"{id_prefix}_pp_inpainting_denoising_neg_prompt",
|
|
label="Inpainting negative prompt use [gender] instead of men or woman",
|
|
)
|
|
with gr.Row():
|
|
samplers_names = [s.name for s in sd_samplers.all_samplers]
|
|
inpainting_sampler = gr.Dropdown(
|
|
choices=samplers_names,
|
|
value=[samplers_names[0]],
|
|
label="Inpainting Sampler",
|
|
elem_id=f"{id_prefix}_pp_inpainting_sampler",
|
|
)
|
|
inpainting_denoising_steps = gr.Slider(
|
|
1,
|
|
150,
|
|
20,
|
|
step=1,
|
|
label="Inpainting steps",
|
|
elem_id=f"{id_prefix}_pp_inpainting_steps",
|
|
)
|
|
|
|
inpaiting_model = gr.Dropdown(
|
|
choices=["Current"] + sd_models.checkpoint_tiles(),
|
|
default="Current",
|
|
label="sd model (experimental)",
|
|
elem_id=f"{id_prefix}_pp_inpainting_sd_model",
|
|
)
|
|
|
|
inpaiting_seed = gr.Number(
|
|
label="Inpainting seed",
|
|
value=0,
|
|
minimum=0,
|
|
precision=0,
|
|
elem_id=f"{id_prefix}_pp_inpainting_seed",
|
|
)
|
|
|
|
gradio_components: List[gr.components.Component] = [
|
|
inpainting_denoising_strength,
|
|
inpainting_denoising_prompt,
|
|
inpainting_denoising_negative_prompt,
|
|
inpainting_denoising_steps,
|
|
inpainting_sampler,
|
|
inpaiting_model,
|
|
inpaiting_seed,
|
|
]
|
|
|
|
for component in gradio_components:
|
|
setattr(component, "do_not_save_to_config", True)
|
|
|
|
return gradio_components
|