Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix typo that prevents training Textual Inversion

Browse Source 
There was a small typo in line 529.

shared.ops.embeddings_train_log caused an attribute not found exception when training TIs.
This commit is contained in:
David Pina
2023-04-29 22:41:24 +02:00
committed by GitHub
parent 3c41056123
commit 2220316920
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/textual_inversion/textual_inversion.py
+1 -1
View File
@@ -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
})