hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-08 13:19:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge branch 'ae'

Browse Source
This commit is contained in:
AUTOMATIC
2022-10-21 13:34:48 +03:00
parent bf30673f51 2362d5f00e
commit 7d6b388d71
9 changed files with 326 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modules/textual_inversion/textual_inversion.py
View File

@@ -276,6 +276,7 @@ def train_embedding(embedding_name, learn_rate, batch_size, data_root, log_direc
loss.backward()
optimizer.step()
epoch_num = embedding.step // len(ds)
epoch_step = embedding.step - (epoch_num * len(ds)) + 1