aboutsummaryrefslogtreecommitdiff
path: root/modules/textual_inversion
diff options
context:
space:
mode:
authorflamelaw <flamelaw.com3d2@gmail.com>2022-11-23 02:49:01 +0900
committerflamelaw <flamelaw.com3d2@gmail.com>2022-11-23 02:49:01 +0900
commit89d8ecff09b426ddc89eb5b432825f8f4c218051 (patch)
tree8e9670a9f985db140b3cde8c7987909f235ba1f6 /modules/textual_inversion
parent5b57f61ba47f8b11d19a5b46e7fb5a52458abae5 (diff)
small fixes
Diffstat (limited to 'modules/textual_inversion')
-rw-r--r--modules/textual_inversion/textual_inversion.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py
index 3036e48a..fee08e33 100644
--- a/modules/textual_inversion/textual_inversion.py
+++ b/modules/textual_inversion/textual_inversion.py
@@ -436,7 +436,7 @@ def train_embedding(embedding_name, learn_rate, batch_size, gradient_step, data_
shared.state.textinfo = f"""
<p>
Loss: {loss_step:.7f}<br/>
-Step: {embedding.step}<br/>
+Step: {steps_done}<br/>
Last prompt: {html.escape(batch.cond_text[0])}<br/>
Last saved embedding: {html.escape(last_saved_file)}<br/>
Last saved image: {html.escape(last_saved_image)}<br/>