diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-13 11:32:15 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-13 11:32:15 +0000 |
| commit | a176d89487d92f5a5b152401e5c424b34ff43b96 (patch) | |
| tree | 914910f9e24f3f489d1a273ec5abf4531dcd502c /modules/textual_inversion/textual_inversion.py | |
| parent | 486bda9b331d054870e2b3551f94ece7aa39574d (diff) | |
| download | stable-diffusion-webui-gfx803-a176d89487d92f5a5b152401e5c424b34ff43b96.tar.gz stable-diffusion-webui-gfx803-a176d89487d92f5a5b152401e5c424b34ff43b96.tar.bz2 stable-diffusion-webui-gfx803-a176d89487d92f5a5b152401e5c424b34ff43b96.zip | |
print bucket sizes for training without resizing images #6620
fix an error when generating a picture with embedding in it
Diffstat (limited to 'modules/textual_inversion/textual_inversion.py')
| -rw-r--r-- | modules/textual_inversion/textual_inversion.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index 853246a6..e23906ca 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -479,7 +479,7 @@ def train_embedding(embedding_name, learn_rate, batch_size, gradient_step, data_ epoch_num = embedding.step // steps_per_epoch
epoch_step = embedding.step % steps_per_epoch
- description = f"Training textual inversion [Epoch {epoch_num}: {epoch_step+1}/{steps_per_epoch}]loss: {loss_step:.7f}"
+ description = f"Training textual inversion [Epoch {epoch_num}: {epoch_step+1}/{steps_per_epoch}] loss: {loss_step:.7f}"
pbar.set_description(description)
shared.state.textinfo = description
if embedding_dir is not None and steps_done % save_embedding_every == 0:
|