aboutsummaryrefslogtreecommitdiffstats
path: root/modules/textual_inversion/textual_inversion.py
diff options
context:
space:
mode:
authorVladimir Mandic <mandic00@live.com>2023-01-03 15:34:51 +0000
committerGitHub <noreply@github.com>2023-01-03 15:34:51 +0000
commit192ddc04d6de0d780f73aa5fbaa8c66cd4642e1c (patch)
treed699777c81a2661449110157ee77ecc19e71a785 /modules/textual_inversion/textual_inversion.py
parent1d9dc48efda2e8da6d13fc62e65500198a9b041c (diff)
downloadstable-diffusion-webui-gfx803-192ddc04d6de0d780f73aa5fbaa8c66cd4642e1c.tar.gz
stable-diffusion-webui-gfx803-192ddc04d6de0d780f73aa5fbaa8c66cd4642e1c.tar.bz2
stable-diffusion-webui-gfx803-192ddc04d6de0d780f73aa5fbaa8c66cd4642e1c.zip
add job info to modules
Diffstat (limited to 'modules/textual_inversion/textual_inversion.py')
-rw-r--r--modules/textual_inversion/textual_inversion.py1
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