From 2220316920cefc3d29d812aaf901d9714a42ec7d Mon Sep 17 00:00:00 2001 From: David Pina Date: Sat, 29 Apr 2023 22:41:24 +0200 Subject: [PATCH] Fix typo that prevents training Textual Inversion There was a small typo in line 529. shared.ops.embeddings_train_log caused an attribute not found exception when training TIs. --- modules/textual_inversion/textual_inversion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index 75bd1cd88..2180d7f32 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -526,7 +526,7 @@ def train_embedding(id_task, embedding_name, learn_rate, batch_size, gradient_st save_embedding(embedding, optimizer, checkpoint, embedding_name_every, last_saved_file, remove_cached_checksum=True) embedding_yet_to_be_embedded = True - write_loss(log_directory, shared.ops.embeddings_train_log, embedding.step, steps_per_epoch, { + write_loss(log_directory, shared.opts.embeddings_train_log, embedding.step, steps_per_epoch, { "loss": f"{loss_step:.7f}", "learn_rate": scheduler.learn_rate })