diff options
| author | Shondoit <shondoit@gmail.com> | 2023-01-12 15:29:00 +0000 |
|---|---|---|
| committer | Shondoit <shondoit@gmail.com> | 2023-02-15 09:03:59 +0000 |
| commit | edb10092de516dda5271130ed53628387780a859 (patch) | |
| tree | d7f78d9a5b238b3836f9811327a4160b59ef14eb /modules/ui.py | |
| parent | bc50936745e1a349afdc28cf1540109ba20bc71a (diff) | |
| download | stable-diffusion-webui-gfx803-edb10092de516dda5271130ed53628387780a859.tar.gz stable-diffusion-webui-gfx803-edb10092de516dda5271130ed53628387780a859.tar.bz2 stable-diffusion-webui-gfx803-edb10092de516dda5271130ed53628387780a859.zip | |
Add ability to choose using weighted loss or not
Diffstat (limited to 'modules/ui.py')
| -rw-r--r-- | modules/ui.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/modules/ui.py b/modules/ui.py index f5df1ffe..efb87c23 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -1191,6 +1191,8 @@ def create_ui(): create_image_every = gr.Number(label='Save an image to log directory every N steps, 0 to disable', value=500, precision=0, elem_id="train_create_image_every")
save_embedding_every = gr.Number(label='Save a copy of embedding to log directory every N steps, 0 to disable', value=500, precision=0, elem_id="train_save_embedding_every")
+ use_weight = gr.Checkbox(label="Use PNG alpha channel as loss weight", value=False, elem_id="use_weight")
+
save_image_with_stored_embedding = gr.Checkbox(label='Save images with embedding in PNG chunks', value=True, elem_id="train_save_image_with_stored_embedding")
preview_from_txt2img = gr.Checkbox(label='Read parameters (prompt, etc...) from txt2img tab when making previews', value=False, elem_id="train_preview_from_txt2img")
@@ -1304,6 +1306,7 @@ def create_ui(): shuffle_tags,
tag_drop_out,
latent_sampling_method,
+ use_weight,
create_image_every,
save_embedding_every,
template_file,
@@ -1337,6 +1340,7 @@ def create_ui(): shuffle_tags,
tag_drop_out,
latent_sampling_method,
+ use_weight,
create_image_every,
save_embedding_every,
template_file,
|