StableDiffusion
/
KohyaSS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix Utilities layout

This commit is contained in:
bmaltais 2022-12-15 18:31:36 -05:00
parent cb98b4f149
commit 515c52e2cf
1 changed files with 14 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
dreambooth_gui.py
View File

@ -477,7 +477,7 @@ def dreambooth_folder_preparation(
else:
training_dir = os.path.join(
util_training_dir_input,
f"img/{int(util_training_images_repeat_input)}_{util_training_images_prompt_input}",
f"img/{int(util_training_images_repeat_input)}_{util_training_images_prompt_input} {util_regularization_images_prompt_input}",
)
# Remove folders if they exist
@ -738,6 +738,18 @@ with interface:
gr.Markdown(
"This utility will create the required folder structure for the training images and regularisation images that is required for kohys_ss Dreambooth method to properly run."
)
with gr.Row():
util_training_images_prompt_input = gr.Textbox(
label="Instance prompt",
placeholder="Eg: asd",
interactive=True,
)
util_regularization_images_prompt_input = gr.Textbox(
label="Class prompt",
placeholder=
"Eg: person",
interactive=True,
)
with gr.Row():
util_training_images_dir_input = gr.Textbox(
label="Training images",
@ -753,16 +765,11 @@ with interface:
value=40,
interactive=True,
elem_id="number_input")
util_training_images_prompt_input = gr.Textbox(
label="Training images prompt",
placeholder="Prompt for the training images. Eg: asd",
interactive=True,
)
with gr.Row():
util_regularization_images_dir_input = gr.Textbox(
label="Regularisation images",
placeholder=
"Directory containing the regularisation images",
"(Optional) Directory containing the regularisation images",
interactive=True,
)
button_util_regularization_images_dir_input = gr.Button(
@ -775,12 +782,6 @@ with interface:
value=1,
interactive=True,
elem_id="number_input")
util_regularization_images_prompt_input = gr.Textbox(
label="Regularisation images class prompt",
placeholder=
"Prompt for the regularisation images. Eg: person",
interactive=True,
)
with gr.Row():
util_training_dir_input = gr.Textbox(
label="Destination training directory",