add job info to modules

This commit is contained in:
Vladimir Mandic
2023-01-03 10:34:51 -05:00
committed by GitHub
parent 1d9dc48efd
commit 192ddc04d6
4 changed files with 16 additions and 4 deletions

View File

@@ -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