diff options
author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-01-04 11:04:50 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-01-04 11:04:50 +0000 |
commit | 545ae8cb1c17a038a5bfd126546acf4b9cab334c (patch) | |
tree | 77d31fa9b40d86415c524aaef80f31123412a522 /modules/textual_inversion/textual_inversion.py | |
parent | a8ad8666cd2b5df525afe702a953095877bcecde (diff) | |
parent | d8d206c1685d1e7027d4af82ed18d106f41d1cc4 (diff) | |
download | stable-diffusion-webui-gfx803-545ae8cb1c17a038a5bfd126546acf4b9cab334c.tar.gz stable-diffusion-webui-gfx803-545ae8cb1c17a038a5bfd126546acf4b9cab334c.tar.bz2 stable-diffusion-webui-gfx803-545ae8cb1c17a038a5bfd126546acf4b9cab334c.zip |
Merge pull request #6264 from vladmandic/add-state-info
add missing state info
Diffstat (limited to 'modules/textual_inversion/textual_inversion.py')
-rw-r--r-- | modules/textual_inversion/textual_inversion.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index fd253477..2c1251d6 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -245,6 +245,7 @@ def train_embedding(embedding_name, learn_rate, batch_size, gradient_step, data_ create_image_every = create_image_every or 0
validate_train_inputs(embedding_name, learn_rate, batch_size, gradient_step, data_root, template_file, steps, save_embedding_every, create_image_every, log_directory, name="embedding")
+ shared.state.job = "train-embedding"
shared.state.textinfo = "Initializing textual inversion training..."
shared.state.job_count = steps
|