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

Merge pull request #628 from dpinav/master

Browse Source 
Fix typo that causes exception when training Textual Inversion
This commit is contained in:
Vladimir Mandic
2023-04-29 17:19:31 -04:00
committed by GitHub
parent 3c41056123 2220316920
commit 25265f592a
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
})