diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-02-19 09:41:35 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-02-19 09:41:35 +0000 |
| commit | e452facef476342af30ab7c0d510f715c6ae9ea0 (patch) | |
| tree | 30b5b4e5a179e9d88e75ac0abdcf0090e1c74397 /modules/ui.py | |
| parent | 43137773227344dd79b56c227dbc9fb54ac7c337 (diff) | |
| parent | dfb3b8f398239876cdfc03657680e50c76a1fed8 (diff) | |
| download | stable-diffusion-webui-gfx803-e452facef476342af30ab7c0d510f715c6ae9ea0.tar.gz stable-diffusion-webui-gfx803-e452facef476342af30ab7c0d510f715c6ae9ea0.tar.bz2 stable-diffusion-webui-gfx803-e452facef476342af30ab7c0d510f715c6ae9ea0.zip | |
Merge pull request #6700 from Shondoit/weighted-learning
Weighted learning of TIs and HNs
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 2fdbda42..0516c643 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,
|